AMD發布ROCm 5.6開放軟體平台 可為AI帶來最佳化效能

AMD發布全新AMD ROCm 5.6開放軟體平台，AMD人工智慧事業群資深副總裁Vamsi Boppana於部落格文章中重點介紹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顯示卡，並將陸續發佈更多支援顯示卡和擴充功能。