ROCm/rocm-systems
super repo for rocm systems projects
ROCm Systems – AMD GPU 计算库的统一超级仓库
它是什么
- 一个将许多独立的 ROCm (Radeon Open Compute) 系统组件(如
hip、rocprofiler、rocm‑smi、rocshmem等)汇集在一起的单体仓库。其目标是使构建、测试这些底层 GPU 库并为其做出贡献变得更加简单和一致。
它对 AI/ML 的重要性
- ROCm 提供了 PyTorch、TensorFlow 和 JAX 等框架在 AMD GPU 上运行所需的驱动级和运行时支持。通过整合各种 ROCm 库,该仓库简化了支撑 AMD 硬件上 GPU 加速训练和推理的软件栈的开发。
关键组件(每个组件都位于 projects/ 目录下,且是一个成熟的库)
| 组件 | 用途 |
|---|---|
hip |
AMD 用于编写 GPU 内核的 CUDA 兼容语言/运行时。 |
rocprofiler‑sdk / rocprofiler‑systems |
用于 ROCm 应用程序的性能分析和性能评估工具。 |
amdsmi / rocm‑smi‑lib |
公开 GPU 健康状况、温度、功耗等的系统管理接口。 |
rocshmem |
用于多 GPU 通信的共享内存 API。 |
roctracer |
用于捕获 GPU 活动的追踪库。 |
rocdecode / rocjpeg |
在 GPU 上运行的媒体处理库(图像/视频解码)。 |
rccl (未在表中列出但属于 ROCm) |
用于多 GPU 训练的集合通信库。 |
hipfile, hipother, hip‑tests |
HIP 生态系统的辅助工具和测试套件。 |
rocdbgapi |
ROCm 内核的调试 API。 |
rocprofiler‑compute, rocprofiler‑register |
额外的性能分析后端和注册辅助工具。 |
CI / 健康状况
- README 显示了一个详细的状态表,其中包含每个组件的持续集成流水线(构建、CodeQL、格式化、清理器等)的徽章。大多数组件被标记为 Completed,这意味着它们已经迁移到超级仓库中,并被视为事实来源。
如何开始
- 克隆仓库并按照
CONTRIBUTING.md中的说明进行操作(稀疏检出、开发环境设置、PR 工作流程)。 - 每个组件都可以使用 ROCm 构建系统 TheRock 单独构建;顶层 CI 也运行多组件测试。
- 对于对特定库(例如
hip或rocprofiler-sdk)感兴趣的开发者,请导航至projects/<component>/并查阅该子项目的 README 以获取构建标志和使用示例。
许可证
- 该仓库在根目录下没有强制执行单一许可证。每个子项目保留其原始许可证(请参阅每个
projects/或shared/目录内的LICENSE*文件)。用户应检查他们打算使用的特定组件的许可证。
谁应该关注
- AMD GPU 开发者:构建或扩展底层计算库。
- 框架维护者(PyTorch、TensorFlow 等):需要为 AMD 硬件提供稳定的 ROCm 软件栈。
- 性能工程师:寻找 AMD GPU 的性能分析、追踪或系统监控工具。
- 贡献者:希望在多个 ROCm 库之间工作而无需管理多个独立仓库。
总结 ROCm Systems 是一个真正的生产级代码库,它整合了 AMD GPU 计算生态系统的基本构建块,直接支持 AMD 硬件上的 AI/ML 工作负载。
相关
- 项目
- 项目
- 项目
- 项目
- 项目