事件

class torch.xpu.Event(enable_timing=False)

XPU 事件的包装器。

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

底层的 XPU 事件在首次记录时延迟初始化。创建后，只有同一设备上的流可以记录事件。但是，任何设备上的流都可以等待该事件。

参数

enable_timing (bool, 可选) – 指示事件是否应测量时间（默认值：False）

elapsed_time(end_event)

返回经过的时间。

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

query()

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

返回值

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

返回类型

bool

record(stream=None)

在给定流中记录事件。

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

synchronize()

等待事件完成。

等待直到此事件当前捕获的所有工作完成。这会阻止 CPU 线程继续执行，直到事件完成。

wait(stream=None)

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

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