事件

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