AMD 針對 AI 應用發布全新 AMD ROCm 5.6 開放軟體平台,AMD 人工智慧事業群資深副總裁 Vamsi Boppana 於部落格文章中重點介紹 ROCm 5.6 的新功能。
AMD ROCm 5.6 版正式發布
ROCm 5.6 為人工智慧 (AI) 與大型語言模型 (LLM) 工作負載帶來:
- 將 Hugging Face 單元測試套件整合至 ROCm QA 中。
- 在 PyTorch 2.0 感應器模式下對 OpenAI Triton 的增量支援。
- 透過 ROCm 對 PyTorch、TensorFlow 和 JAX 提供 OpenXLA 支援,支援更廣泛的社群。
對於 HPC 應用程式,ROCm 5.6 引入對快速傅立葉變換 (Fast Fourier Transform,FFT)、基礎線性代數程式集 (Basic Linear Algebra Subprograms,BLAS) 與計算機 (solver) 等多個數學函式庫的效能提升。這些數學函式庫是 HPC 應用程式的基礎,並增強了 ROCm 開發和部署工具,包括安裝、ROCgdb (CPU-GPU 整合除錯器)、ROCm 分析工具和文件。
為提供最卓越的效能,AMD 也將於未來數月內對框架和後端編譯器作進一步的最佳化,其中包括 MLIR 基礎架構改進,鞏固 AMD 對 OpenAI Triton 與 OpenXLA 編譯器的支援。客戶將在 Hugging Face 的 AMD Hub 上看到更多為 AMD 解決方案作出最佳化的開源 AI 模型。
此外,AMD 將於今年稍後正式在採用 AMD RDNA 3 架構的特定顯示卡上釋出 AMD ROCm 支援。Linux 預計將於今年秋季正式支援基於 RDNA 3 架構的顯示卡,首先是 48GB Radeon PRO W7900 繪圖卡與 24GB Radeon RX 7900 XTX 顯示卡,並將陸續發佈更多支援顯示卡和擴充功能。