Event

class torch.xpu.Event(enable_timing=False)

XPU 事件的包裝函式。

XPU 事件是同步標記，可用於監控裝置的進度以及同步 XPU 串流。

底層的 XPU 事件會在首次記錄事件時延遲初始化。建立之後，只有在相同裝置上的串流才能記錄事件。但是，任何裝置上的串流都可以等待該事件。

參數

enable_timing (bool, optional) – 指示事件是否應測量時間 (預設值：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()。