ROCm/TheRock

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

What it solves

TheRock simplifies the installation and build process for HIP and ROCm, removing the complexity of traditional package-based installations. It provides a lightweight build platform that allows developers and researchers to access the latest ROCm capabilities, including nightly releases of ROCm, PyTorch, and JAX.

How it works

It functions as a CMake super-project that manages the source builds of HIP and ROCm. It allows users to customize their build via feature flags (e.g., enabling specific math or ML libraries) and supports multiple Linux distributions and native Windows. The project also includes tools for developing individual ROCm components and comprehensive CI/CD pipelines for testing and releasing.

Who it’s for

  • ROCm contributors
  • Developers and researchers needing the latest ROCm features
  • Advanced users who want to avoid package-based installation complexity

Highlights

  • Nightly releases for ROCm, PyTorch, and JAX
  • Support for building PyTorch and JAX from source with ROCm
  • Cross-platform support for multiple Linux distributions and native Windows
  • Highly customizable build configuration via CMake flags
  • Integrated support for ccache to accelerate frequent rebuilds

Related

  • Project
    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.

  • Project
    ROCm/rocm-libraries

    A consolidated super-repository for ROCm libraries that streamlines the development, build, and integration of high-performance compute components for AMD GPUs.

  • Dispatch
    Easily Build and Share ROCm Kernels with Hugging Face

    Hugging Face introduces a streamlined workflow using the kernels library and kernel-builder to build, package, and share ROCm-compatible GPU kernels on the Hugging Face Hub.

  • Project
    ROCm/legacy-rocm-build

    An open-source software stack of drivers, tools, and APIs that enables GPU programming and AI acceleration on AMD hardware.

  • Project
    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.