Event

class torch.xpu.Event(enable_timing=False)

XPU事件的包装器。

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

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

参数

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

elapsed_time(end_event)

返回经过的时间。

报告的时间以毫秒为单位，在记录事件和记录end_event之间。

query()

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

返回

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

返回类型

布尔值

record(stream=None)

在给定的流中记录事件。

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

synchronize()

等待事件完成。

Waits until the completion of all work currently captured in this event. This prevents the CPU thread from proceeding until the event completes.

wait(stream=None)

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

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