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()。