事件

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

MPS事件的包装器。

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

Parameters

enable_timing (bool, 可选) – 指示事件是否应测量时间 (默认值: False)

elapsed_time(end_event)

返回事件记录后到end_event记录前的毫秒数。

query()

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

record()

在默认流中记录事件。

synchronize()

等待当前事件中捕获的所有工作完成。 这会阻止CPU线程继续执行，直到事件完成。

wait()

使所有提交到默认流的未来工作等待此事件。