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