活动 ¶

class torch.xpu.Event(enable_timing=False)

XPU 事件包装器。

XPU 事件是同步标记，可用于监控设备的进度，以及同步 XPU 流。

基础 XPU 事件在事件首次记录时进行懒加载初始化。创建后，只有同一设备上的流可以记录事件。然而，任何设备上的流都可以等待事件。

参数:

enable_timing（布尔值，可选）- 表示事件是否测量时间（默认： False ）

elapsed_time(end_event)[source][source]

返回经过的时间。

报告的时间是在事件记录后和 end_event 记录前以毫秒为单位的。

query()

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

返回值:

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

返回类型:

布尔型

record(stream=None)

在给定的流中记录事件。

如果未指定流，则使用 torch.xpu.current_stream() 。流的设备必须与事件设备匹配。

synchronize()

等待事件完成。

等待此事件中当前捕获的所有工作完成。这可以防止 CPU 线程在事件完成之前继续执行。

wait(stream=None)[源][源] ¶

使所有提交给指定流的未来工作等待此事件。

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