torch.mps.profiler.start¶
- torch.mps.profiler.start(mode='interval', wait_until_completed=False)[source][source]¶
從 MPS 後端開始 OS Signpost 跟蹤。
生成的 OS Signposts 可以在 XCode Instruments 日誌工具中記錄和檢視。
- 引數
mode (str) – OS Signpost 跟蹤模式可以是“interval”、“event”或兩者兼有“interval,event”。 interval 模式跟蹤操作的執行時長，而 event 模式標記執行完成。有關更多資訊，請參閱文件 Recording Performance Data。
wait_until_completed (bool) – 等待直到 MPS Stream 完成執行每個編碼的 GPU 操作。這有助於在跟蹤時間線上生成單個分派。請注意，啟用此選項會對效能產生負面影響。
