评价此页

torch.mps.profiler.start#

torch.mps.profiler.start(mode='interval', wait_until_completed=False)[source]#

开始MPS后端中的OS Signpost跟踪。

生成的OS Signpost可以被XCode Instruments Logging工具记录和查看。

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

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