ROCm/TheRock

The HIP Environment and ROCm Kit - A lightweight open source build system for HIP and ROCm

解决的问题

TheRock 简化了 HIP 和 ROCm 的安装与构建流程，消除了传统基于包的安装的复杂性。它提供了一个轻量级的构建平台，使开发者和研究人员能够访问最新的 ROCm 功能，包括 ROCm、PyTorch 和 JAX 的夜间发布版本。

工作原理

它作为一个 CMake 超项目运行，负责管理 HIP 和 ROCm 的源码构建。用户可以通过功能标志（例如启用特定的数学或机器学习库）自定义构建，并支持多种 Linux 发行版和原生 Windows。该项目还包含用于开发单个 ROCm 组件的工具，以及用于测试和发布的全面 CI/CD 流水线。

适用人群

  • ROCm 贡献者
  • 需要最新 ROCm 功能的开发者和研究人员
  • 希望避免基于包安装复杂性的高级用户

主要亮点

  • ROCm、PyTorch 和 JAX 的夜间发布版本
  • 支持从源码构建 PyTorch 和 JAX（基于 ROCm）
  • 支持多种 Linux 发行版和原生 Windows 的跨平台构建
  • 通过 CMake 标志实现高度可定制的构建配置
  • 集成 ccache 以加速频繁的重新构建

相关

  • 项目
    ROCm/rocm-systems

    A consolidated super-repository for ROCm systems projects that streamlines the development, CI, and integration of libraries and tools required for building PyTorch.

  • 项目
    ROCm/rocm-libraries

    一个整合 ROCm 库的超级仓库，旨在简化 AMD GPU 高性能计算组件的开发、构建与集成。

  • Dispatch
    轻松使用 Hugging Face 构建和共享 ROCm 内核

    Hugging Face 介绍了一种使用 kernels 库和 kernel-builder 的简化工作流程，以在 Hugging Face Hub 上构建、打包和共享 ROCm 兼容的 GPU 内核。

  • 项目
    ROCm/legacy-rocm-build

    一个由驱动程序、工具和 API 组成的开源软件栈，使 AMD 硬件上的 GPU 编程和 AI 加速成为可能。

  • 项目
    ROCm/rocm-examples

    A collection of examples and tutorials for the ROCm platform, providing code for HIP runtime functionality, GPU-accelerated applications, and AI model execution.