torch.mps.profiler.profile¶ torch.mps.profiler.profile(mode='interval', wait_until_completed=False)[來源][來源]¶ Context Manager 用於從 MPS 後端啟用產生 OS Signpost 追蹤。 參數 mode (str) – OS Signpost 追蹤模式可以是 “interval”、“event” 或 “interval,event” 兩者。 interval 模式追蹤操作執行的持續時間，而 event 模式標記執行的完成。 更多資訊請參閱文件 記錄效能資料。 wait_until_completed (bool) – 等待 MPS Stream 完成執行每個編碼的 GPU 操作。 這有助於在追蹤的時間軸上產生單個調度。 請注意，啟用此選項會對效能產生負面影響。
