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,这意味着它们已经迁移到超级仓库中,并被视为事实来源。

如何开始

  1. 克隆仓库并按照 CONTRIBUTING.md 中的说明进行操作(稀疏检出、开发环境设置、PR 工作流程)。
  2. 每个组件都可以使用 ROCm 构建系统 TheRock 单独构建;顶层 CI 也运行多组件测试。
  3. 对于对特定库(例如 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 工作负载。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目