torch.mps.profiler.start

torch.mps.profiler.start(mode='interval', wait_until_completed=False)[源代码]

从MPS后端开始OS踪迹追踪。

生成的 OS Signposts 可以使用 XCode Instruments Logging 工具进行记录和查看。

参数

• mode (str) – OS Signpost 跟踪模式可以是 “interval”，“event” 或者同时为 “interval,event”。其中，“interval” 模式记录操作执行的持续时间，而 “event” 模式则标记执行完成。更多详情请参阅文档 记录性能数据。

• wait_until_completed (bool) – 等待 MPS Stream 完成每个编码的 GPU 操作。这有助于在时间线上生成单个调度记录。需要注意的是，启用此选项会降低性能。