Shortcuts

torch.mps.profiler.start

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

从MPS后端开始OS Signpost跟踪。

生成的操作系统标志可以在XCode Instruments日志工具中记录和查看。

Parameters
  • 模式 (字符串) – OS Signpost 跟踪模式可以是“间隔”、“事件”, 或者两者“间隔,事件”。 间隔模式跟踪操作的执行持续时间, 而事件模式标记执行的完成。 有关更多信息,请参阅文档 记录性能数据

  • wait_until_completed (bool) – 等待直到MPS流完成执行每个编码的GPU操作。这有助于在跟踪的时间线上生成单个调度。 请注意,启用此选项会对性能产生负面影响。

优云智算