流¶
- class torch.cuda.Stream(device=None, priority=0, **kwargs)[source]¶
CUDA 流的包装器。
CUDA 流是属于特定设备的线性执行序列,独立于其他流。有关详细信息,请参阅CUDA 语义。
- 参数
device (torch.device or int, optional) – 分配流的设备。如果
device
为None
(默认值)或负整数,则将使用当前设备。priority (int, optional) – 流的优先级,应为 0 或负数,负数表示更高的优先级。默认情况下,流的优先级为 0。
- record_event(event=None)[source]¶
记录事件。
- 参数
event (torch.cuda.Event, optional) – 要记录的事件。如果未提供,将分配一个新的事件。
- 返回值
已记录的事件。
- wait_event(event)[source]¶
使提交到流的所有未来工作等待事件。
- 参数
event (torch.cuda.Event) – 要等待的事件。