英特爾(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。
留言 0