ROCm/legacy-rocm-build

解決的問題

ROCm 提供一個開源軟體堆疊，讓開發者能為高階運算（HPC）、人工智慧（AI）、科學計算與電腦輔助設計（CAD）程式設計 AMD GPU。它彌補了底層 GPU 核心與終端使用者應用程式之間的差距，使複雜工作負載得以實現 GPU 加速。

工作原理

ROCm 由 HIP 驅動，HIP 是一種 C++ 執行階段 API 與核心語言，讓開發者能使用類似 NVIDIA CUDA 的介面撰寫可移植的 GPU 程式碼。該堆疊包含全面的驅動程式、開發工具與 API，包括：

• 數學與運算函式庫：專為線性代數、FFT 與隨機數產生設計的函式庫（例如 rocBLAS、rocFFT）。
• 執行階段與編譯器：如 LLVM 與 ROCr 執行階段，用於建構與執行 GPU 工作負載。
• 整合：與 PyTorch 與 TensorFlow 等主要機器學習框架直接相容。
• 工具：專用的效能分析、除錯與監控工具（例如 ROCgdb、AMD SMI）。

適用對象

專為使用 AMD GPU 硬體進行 AI 訓練與推論、科學計算與高階運算的開發者與研究人員設計。

主要亮點

• 類 CUDA 介面：使用 HIP 實現可移植的 GPU 程式設計。
• 豐富的函式庫生態系：包含大量數學、通訊與媒體函式庫。
• 主流框架支援：與 PyTorch、TensorFlow 與 JAX 集成。
• 廣泛的硬體支援：相容於多種 Linux 發行版中的 CDNA 與 RDNA 架構。