快捷方式

事件

class torch.mps.event.Event(enable_timing=False)[source]

MPS 事件的包装器。

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

参数

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

elapsed_time(end_event)[source]

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

query()[source]

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

record()[source]

在默认流中记录事件。

synchronize()[source]

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

wait()[source]

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源