流

class torch.xpu.Stream(device=None, priority=0, **kwargs)

XPU 流的包装器。

XPU 流是属于特定设备的线性执行序列，独立于其他流。它支持 with 语句作为上下文管理器，以确保 with 块内的操作符在相应的流上运行。

参数

• device (torch.device 或 int, 可选) – 分配流的设备。如果 device 是 None（默认值）或负整数，将使用当前设备。

• priority (int, 可选) – 流的优先级，可以是正数、0 或负数。数值越低表示优先级越高。默认情况下，优先级设置为 0。如果值超出允许的优先级范围，它将被自动映射到最近的有效优先级（大正数的最低优先级或大负数的最高优先级）。

query()

检查所有提交的工作是否已完成。

返回值

一个布尔值，指示此流中的所有内核是否已完成。

返回类型

bool

record_event(event=None)

记录一个事件。

参数

event (torch.xpu.Event, 可选) – 要记录的事件。如果未给定，将分配一个新的。

返回值

记录的事件。

synchronize()

等待此流中的所有内核完成。

wait_event(event)

使提交到此流的所有后续工作等待一个事件。

参数

event (torch.xpu.Event) – 要等待的事件。

wait_stream(stream)

与另一个流同步。

提交到此流的所有后续工作将等待，直到调用时提交到给定流的所有内核完成。

参数

stream (Stream) – 要同步的流。