事件

类torch.mps.event.Event(enable_timing=False)[源代码]

MPS事件的封装类。

MPS事件是同步标记，可以用来监控设备的进度，精确测量时间，并同步MPS流。

参数

enable_timing (bool, 可选) – 是否启用事件的时间测量（默认值: False）

elapsed_time(end_event)[源代码]

返回从事件记录到结束事件记录之间经过的时间（以毫秒为单位）。

query()[源代码]

如果事件中所有已捕获的工作都已完成，则返回 True。

record()[源代码]

将事件记录到默认流中。

synchronize()[源代码]

等待此事件中所有当前工作的完成。这会阻止CPU线程在此事件完成前继续执行。

wait()[源代码]

让所有提交到默认流的 future 工作都等待此事件。