ROCm

定义

ROCm(Radeon Open Compute)是 AMD 于 2016 年推出的开源 GPU 计算平台,包含驱动、运行时、编译器、数学库与深度学习库,定位对标 NVIDIA 的 CUDA 生态,支撑其 Instinct 系列数据中心 GPU。

关键组件

• HIP:C++ 编程接口,API 与 CUDA 高度对应,提供 hipify 工具把 CUDA 源码自动转换为可在 AMD 与 NVIDIA GPU 上双平台编译的代码。
• rocBLAS/MIOpen/RCCL:分别对应 cuBLAS、cuDNN、NCCL 的功能角色。
• 基于 LLVM 的开源编译器栈。

现状

PyTorch 等主流框架已提供官方 ROCm 后端,大型云厂商和实验室在 MI 系列加速器上开展大模型训练与推理。相较 CUDA,ROCm 的短板主要在消费级显卡支持范围、文档与长尾算子的成熟度,但其完全开源的特性吸引了希望摆脱单一供应商锁定的用户。

意义

ROCm 是目前对 CUDA 生态最具实质竞争力的开放替代方案之一。