Event

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

MPS 事件的封裝器。

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

參數

enable_timing (bool, optional) – 指示事件是否應該測量時間 (預設值： False)

elapsed_time(end_event)

回傳事件被記錄後，到 end_event 被記錄前所經過的時間，單位為毫秒。

query()

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

record()

在預設的串流中記錄事件。

synchronize()

等待目前事件中捕獲的所有工作完成。這可以防止 CPU 線程在事件完成之前繼續執行。

wait()

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