近日，AMD 發布了新的 AMD ROCm 5.6 開放軟件平臺，AMD AI群組高級副總裁 Vamsi Boppana 在博客中重點介紹了 ROCm 5.6 的一些新功能。

對于 AI 和 LLM 工作負載，ROCm 5.6 帶來了：

· 將 Hugging Face 單元測試套件集成到 ROCm QA 中

· PyTorch 2.0 感應器模式下對 OpenAI Triton 的增量支持

· 通過 ROCm 對 PyTorch、TensorFlow 和 JAX 提供 OpenXLA 支持，支持更廣泛的社區

對于 HPC 應用程序，ROCm 5.6 引入了對 FFT、BLAS 和求解器等多個數學庫的改進，這些數學庫構成了 HPC 應用程序的基礎，并增強了 ROCm 開發和部署工具，包括安裝、ROCgdb（CPU-GPU 集成調試器）、ROCm 分析器 和文檔。

在接下來的幾個月中，AMD 還將進一步優化框架和后端編譯器，以實現更佳性能，其中包括 MLIR 基礎設施改進，鞏固 AMD 對 OpenAI Triton 和 OpenXLA 編譯器的支持。客戶還可以在 Hugging Face 的 AMD 中心內期待更多針對 AMD 解決方案優化的開源 AI 模型。

此外，AMD 透露，今年晚些時候，對AMD ROCm 的正式支持將開始在部分基于 AMD RDNA 3 的 GPU 上推出。Linux 上對基于 RDNA 3 的 GPU 的正式支持計劃于今年秋季開始推出，首先是 48GB Radeon PRO W7900 和 24GB Radeon RX 7900 XTX，隨著時間的推移，還會發布更多卡和擴展功能。

責任編輯：彭菁