torch.mps.profiler.start¶
- torch.mps.profiler.start(mode='interval', wait_until_completed=False)[source][source]¶
從 MPS 後端啟動 OS Signpost 追蹤。
產生的 OS Signposts 可以在 XCode Instruments Logging 工具中記錄和檢視。
- 參數
mode (str) – OS Signpost 追蹤模式可以是 “interval”、”event”，或兩者皆有 “interval,event”。interval 模式追蹤操作的執行期間，而 event 模式標記執行的完成。詳情請參閱文件 Recording Performance Data。
wait_until_completed (bool) – 等待 MPS Stream 完成執行每個編碼的 GPU 操作。這有助於在追蹤的時間軸上產生單個調度。請注意，啟用此選項會對效能產生負面影響。
