torch.mps.profiler.profile

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

一个上下文管理器，用于从MPS后端启用操作系统的Signpost追踪生成。

参数

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

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