流#
- class torch.xpu.Stream(device=None, priority=0, **kwargs)[source]#
XPU流的包装器。
XPU 流是属于特定设备的独立执行的线性序列。它支持 with 语句作为上下文管理器,以确保 with 块内的操作在相应的流上运行。
- 参数
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) – 要等待的事件。