torch.mps.profiler.profile#
- torch.mps.profiler.profile(mode='interval', wait_until_completed=False)[source]#
用於啟用從 MPS 後端生成 OS Signpost 跟蹤的上下文管理器。
- 引數
mode (str) – OS Signpost 跟蹤模式可以是“interval”、“event”或兩者都是“interval,event”。interval 模式跟蹤操作的執行持續時間，而 event 模式標記執行的完成。有關更多資訊，請參閱文件 Recording Performance Data。
wait_until_completed (bool) – 等待 MPS Stream 完成執行每個編碼的 GPU 操作。這有助於在跟蹤的時間線上生成單個分派。請注意，啟用此選項會產生負面性能影響。
