四十年對手變隊友:AMD 與 Intel 把 PyTorch 的 x86 加速收進同一個函式庫
兩家公司 10 月 20 日宣布以開源 oneDNN 作為 PyTorch 在 x86 處理器的統一加速路徑,AMD 把 ZenDNN 併入 oneDNN,並加入主導該函式庫的 UXL 基金會。
在 x86 處理器市場纏鬥四十年的 AMD 與 Intel,10 月 20 日聯手做了一件對開發者更實惠的事:把各自為 PyTorch 做的處理器最佳化,收進同一個開源深度學習函式庫 oneDNN。從此以後,在 x86 CPU 上跑 AI 推論,不必再為不同廠牌的處理器維護兩套加速路徑。
兩套函式庫的稅,開發者繳了很久
問題的根源是歷史包袱。oneDNN 最早是 Intel 為自家處理器打磨的深度學習基本運算集合,負責矩陣乘法、卷積等模型執行時最吃重的底層工作,PyTorch、TensorFlow 等框架都仰賴它。但它對 Intel 晶片的調校最到位,AMD 因此另起 ZenDNN 專案,用同一套介面、換上針對 Zen 核心寫的運算核心。
結果是:同一個模型部署到混合機隊時,開發者得操心底層到底裝了哪一家的函式庫版本,兩條幾乎相同卻又不完全相同的程式碼路徑,也成了長期維護負擔。這次合作就是要把這筆稅免掉。
協議內容:ZenDNN 併入,oneDNN 成預設路徑
根據兩家公司的新聞稿,AMD 與 Intel 已把 oneDNN 定為 PyTorch 內 x86 加速的預設開源函式庫。AMD 將把 ZenDNN 的最佳化成果整合進 oneDNN,同時以指導成員身分加入負責治理 oneDNN 的統一加速基金會(UXL Foundation),從一般貢獻者升格為主導治理的一員。
這項合作來自 x86 生態系諮詢小組的建議。AMD 方面表示,目標是讓開發者更容易打造高效能 AI 應用;Intel 方面則說,開發者不該為了榨出處理器效能,還得分心處理兩條軟體路徑。雙方也計畫透過未來的 oneDNN 更新,把專為矩陣乘法加速設計的下一代 x86 指令集「AI Compute Extensions」帶進 PyTorch,但尚未公布時程。
為什麼是現在:Arm 陣營正在逼近
時間點值得玩味。輝達的 RTX Spark 筆電剛上市,把自家 AI 軟體堆疊帶進消費級 PC;蘋果的 Mac mini 與 Mac Studio 也成為本地跑 AI 模型的熱門選擇。x86 兩大廠若繼續在軟體層各自為政,只會讓開發者更有理由轉向整合度更高的對手。
對企業用戶來說,這次合作的實際意義不在於某個跑分數字,而是部署彈性:同一份程式碼、同一套函式庫,就能在 AMD 與 Intel 的伺服器與 PC 上取得各自的最佳化版本。至於成效如何,要等 ZenDNN 成果真正合併上游、開發者實測之後才見分曉。
資料來源:AMD Newsroom、Intel Newsroom