事件

类torch.xpu.Event(enable_timing=False)[源代码]

XPU事件的封装类。

XPU 事件是用于监控设备进度和同步 XPU 流的同步标记。

当事件首次被记录时，底层的XPU事件会懒惰地进行初始化。创建后，只有在同一设备上的流才能记录该事件，但任何设备上的流都可以等待该事件。

参数

enable_timing (bool, 可选) – 是否启用事件的时间测量（默认值: False）

elapsed_time(end_event)[源代码]

返回 elapsed 时间

时间以毫秒为单位报告，从事件被记录之后到结束事件被记录之前。

query()[源代码]

检查所有由事件捕获的工作是否已经完成。

返回值

一个布尔值，表示事件中所有已捕获的工作是否已完成。

返回类型

bool

record(stream=None)[源代码]

将事件记录到指定的流中。

如果没有指定流，则使用 torch.xpu.current_stream()。此外，流所在的设备必须与事件所在的设备一致。

synchronize()[源代码]

等待事件结束。

等待此事件中所有当前工作的完成。这会阻止CPU线程在此事件完成前继续执行。

wait(stream=None)[源代码]

让所有提交到该流的 future 工作都等待此事件。

如果没有指定流，请使用 torch.xpu.current_stream()。