其他 發布日期:
AMD 與 Intel 聯手:PyTorch 的 x86 CPU 加速統一走 oneDNN

發生什麼事
AMD 與 Intel 將兩家在 PyTorch 上的 x86 CPU 最佳化工作整合到 oneDNN,這是由 UXL Foundation 管理的開源函式庫,雙方也同意讓它成為 PyTorch 內預設的 x86 開源加速函式庫。AMD 會把自家 ZenDNN 的最佳化併入 oneDNN,並以指導會員(steering member)身分加入 UXL Foundation。兩家公司也合作開發 AI Compute Extensions,一套專門加速矩陣乘法的新一代 x86 指令集,之後會透過 oneDNN 更新支援 PyTorch。這次公告沒有提到上市時程、具體處理器型號或效能數字。
對地端使用者的意義:在 AMD 或 Intel CPU 上用 PyTorch 跑推論的人,之後只需面對一條最佳化路徑,不必再依廠牌分兩套。
三種角度看這則新聞
市場怎麼看
兩個長年對手願意共用同一層軟體,看得出它們認為一起讓 x86 在 AI 上保有吸引力、對抗 Arm 晶片與 GPU,比在函式庫上互相競爭更划算。這也可能讓純 CPU 伺服器在輕量推論上更容易被選用,不過較大的模型大概仍會以 GPU 為主。
白話版
做出大部分電腦處理器的 AMD 和 Intel,同意把各自替熱門 AI 工具 PyTorch 做的加速技巧,放進同一個共用套件。就像兩家車廠同意用同一種充電插頭,開發者不用再為每個廠牌各做一次。
業內觀點
如果你現在用 EPYC 或 Xeon 跑 PyTorch 模型,請留意哪一版 PyTorch 與 oneDNN 納入 ZenDNN 的成果,再和目前的設定重新跑基準測試,共用函式庫不代表兩家的速度一樣。AI Compute Extensions 需要新硬體和未來的 oneDNN 更新,現有伺服器暫時不會有變化。
訂閱小芽週報
每週一封:本週重點新聞與小芽洞察摘要。隨時可以取消。
來源
閱讀官方原文:AMD Newsroom: AMD and Intel collaborate to simplify AI development on x86 CPUs using PyTorch ↗摘要由地端世界依官方來源撰寫,附上原文連結,不轉載原文。