英特爾宣布攜手AMD,將雙方針對PyTorch的x86 CPU優化成果,整合到開源深度學習庫oneDNN中。這次合作將為PyTorch社區提供一條適用於現代處理器的統一路徑,簡化了跨x86 CPU的開發流程,無需為了充分發揮x86 CPU的性能而採用不同的軟體路徑。

oneDNN由統一加速基金會(UXL)管理,旨在提升AI工作負載在CPU上的運行效率。這一合作將為開發者提供一條通用軟體路徑,優化英特爾和AMD CPU上的AI工作負載,減少x86 AI生態的軟體碎片化,進一步簡化PyTorch應用在x86系統上的開發、維護和部署。
目前英特爾和AMD已就oneDNN達成一致,將其認定為PyTorch內x86加速的默認開源庫。AMD正將ZenDNN優化整合進oneDNN,並以指導委員會成員的身份加入UXL基金會,通過開放協作,推進這項工作的進一步開展。
未來開發者可以通過一個統一的入口,獲得面向英特爾和AMD處理器的PyTorch CPU優化,從而在不同的x86平台上獲得更加一致的開發體驗。英特爾和AMD圍繞oneDNN的共同投入,為現有及未來的x86 AI能力提供了更為完善的支持。
雙方還打算在AI計算擴展(AI Compute Extensions)方面展開合作,通過oneDNN更新,未來進一步集成至PyTorch中。這是面向x86設計的下一代指令集架構,能夠加速矩陣乘法。






