活动 ¶

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

MPS 事件包装器。

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

参数:

enable_timing（布尔值，可选）- 表示事件是否测量时间（默认： False ）

elapsed_time(end_event)[source][source]

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

返回类型:

浮点数

query()

当前事件捕获的所有工作是否已完成

返回类型:

布尔型

record()[source][source]

将事件记录在默认流中

synchronize()

等待此事件中当前捕获的所有工作完成。这可以防止 CPU 线程在事件完成之前继续执行。

wait()

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