torch.mps.profiler.start#
- torch.mps.profiler.start(mode='interval', wait_until_completed=False)[原始碼]#
從 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 操作。這有助於在時間線上生成單一的排程。請注意，啟用此選項會負面影響效能。
