事件

class torch.mps.event.Event(enable_timing=False)

MPS 事件的包裝器。

MPS 事件是同步標記，可用於監控裝置進度、精確測量時間以及同步 MPS 流。

引數

enable_timing (bool, 可選) – 指示事件是否應測量時間（預設值：False）

elapsed_time(end_event)

返回事件記錄後到 end_event 記錄前經過的時間（毫秒）。

返回型別

float

query()

如果事件當前捕獲的所有工作已完成，則返回 True。

返回型別

bool

record()

在預設流中記錄事件。

synchronize()

等待此事件當前捕獲的所有工作完成。這會阻止 CPU 執行緒繼續執行，直到事件完成。

wait()

使提交到預設流的所有未來工作等待此事件。