torch.compiler API 參考¶
如需 torch.compiler
的快速概述,請參閱torch.compiler。
請參閱 |
|
此函數會清除所有編譯快取,並將系統還原至初始狀態。 |
|
告訴編譯器前端 (Dynamo) 跳過對函數的符號內省,而是在遇到函數時直接將其寫入圖表中。 |
|
為函數註冊一個 polyfill 處理程序,通常是來自 C 擴展的 C 函數,以便在圖表中內聯原始函數時取代原始函數。 |
|
此函式用於將函式 fn 標記為具有恆定結果。 |
|
傳回可傳遞給 torch.compile(..., backend="name") 的有效字串。 |
|
此函式提供一個裝飾器來停用函式的編譯。 它還提供遞迴停用所呼叫函式的選項。 |
|
設定編譯器的目前姿態 (stance)。 |
|
指示推論或訓練的新迭代即將開始。 |
|
指示圖形是否作為 torch.compile() 或 torch.export() 的一部分執行/追蹤。 |
|
指示圖形是否透過 TorchDynamo 追蹤。 |