Back to top
  • 공유 分享
  • 인쇄 列印
  • 글자크기 字體大小
已複製網址

英特爾與AMD攜手,將PyTorch的x86 CPU優化路徑整併為oneDNN

電路板上的中央處理器晶片封裝。/ TokenPost.ai

英特爾(INTC)與超微半導體(AMD)宣布,將在AI開發工具PyTorch中,把支援兩家公司x86中央處理器(CPU)的軟體路徑統一整併為oneDNN。這項合作的目的,是減輕開發者為不同CPU廠商分別維護優化路徑的負擔。

英特爾在20日的發布文中表示,依據x86生態系顧問小組(x86 Ecosystem Advisory Group)的建議,雙方同意將oneDNN列為PyTorch中x86加速的預設開源函式庫。AMD將把自家的ZenDNN優化整合進oneDNN,同時以指導成員(Steering Member)身分加入負責管理oneDNN的UXL基金會(UXL Foundation)。

PyTorch是用來開發及執行AI模型的開源框架。oneDNN則是協助CPU與GPU高效處理矩陣乘法、卷積等深度學習運算的函式庫,專案文件指出,除了PyTorch之外,TensorFlow、OpenVINO與ONNX Runtime也都使用這個函式庫。ZenDNN是AMD針對自家Zen處理器的AI運算所開發的函式庫。

英特爾說明,把AMD的優化成果併入oneDNN後,開發者在開發、維護與部署同時支援英特爾與AMD CPU的AI程式時,可以簡化軟體路徑。不過這只是合作目標,此次發布並未包含效能提升的具體數字,也沒有公布整合完成的時程。

oneDNN本身也不會因此變成x86專用。專案文件明確列出,Arm 64位元CPU與RISC-V等其他架構同樣在支援範圍內。此次發布的範圍,僅限於在PyTorch中把英特爾與AMD的x86 CPU優化整合到共同基礎上。

合作對象是CPU,這一點和以GPU為主軸的AI競爭不太一樣。輝達(NVDA)在9月8日的技術部落格中公開了兩條能以Rust撰寫GPU核心的開發路徑,而這次的發布則是針對CPU端的軟體基礎進行整備。

此外,雙方也透過x86生態系顧問小組,就下一代x86指令集擴充AI Compute Extensions(ACE)進行合作,這項擴充是為了加速AI運算中的矩陣乘法。英特爾表示,未來將透過oneDNN的更新,在PyTorch中支援ACE。

<版權所有 ⓒ TokenPost,未經授權禁止轉載與散佈>

最受歡迎

其他相關文章

留言 0

留言小技巧

好文章。 希望有後續報導。 分析得很棒。

0/1000

留言小技巧

好文章。 希望有後續報導。 分析得很棒。
1