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