TheRock 项目最佳实践指南

1. 项目介绍

TheRock 是一个轻量级开源构建系统，专为 HIP（一个从 CUDA 迁移到 ROCm 的工具）和 ROCm（Radeon Open Compute）设计。该项目目前处于早期预览状态，但正在积极开发中，并欢迎贡献者加入。TheRock 提供了包括 CMake 超级项目、ROCm 组件开发工具、PyTorch 与 ROCm 源码构建支持、JAX 支持，以及全面的 CI/CD 管道等功能。

2. 项目快速启动

以下是快速启动 TheRock 项目的步骤：

设置 - Ubuntu (24.04)

1. 安装 Ubuntu 依赖：

   sudo apt install gfortran git git-lfs ninja-build cmake g++ pkg-config xxd patchelf automake python3-venv python3-dev libegl1-mesa-dev
   

2. 克隆仓库：

   git clone https://github.com/ROCm/TheRock.git
   cd TheRock
   

3. 初始化 Python 虚拟环境并安装 Python 依赖：

   python3 -m venv .venv
   source .venv/bin/activate
   pip install -r requirements.txt
   

4. 下载子模块并应用修补：

   python ./build_tools/fetch_sources.py
   

设置 - Windows 11 (VS 2022)

1. 按照 )

Windows 支持指南安装必要的工具。

2. 从 https://github.com/nod-ai/amdgpu-windows-interop 克隆 interop 库。

3. 克隆 TheRock 仓库，初始化 Python 虚拟环境并安装 Python 依赖：

   git clone https://github.com/ROCm/TheRock.git
   cd TheRock
   python -m venv .venv
   .venv\Scripts\Activate.bat
   pip install -r requirements.txt
   

4. 下载子模块并应用修补：

   python ./build_tools/fetch_sources.py
   

3. 应用案例和最佳实践

在 TheRock 中构建 HIP 和 ROCm 的最佳实践包括：

• 使用 CMake 超级项目结构来组织和构建 HIP 和 ROCm 源码。
• 利用项目提供的工具开发单个 ROCm 组件。
• 根据 TheRock 的支持状态，为 Linux 和 Windows 平台构建 PyTorch 与 ROCm 的源码。
• 在 CI/CD 管道中集成构建、测试和发布支持组件的流程。

4. 典型生态项目

TheRock 项目的生态系统中，以下是一些典型的相关项目：

• ROCm: Radeon Open Compute 的官方仓库。
• HIP: 用于从 CUDA 迁移到 ROCm 的工具。
• PyTorch: 一个流行的深度学习框架，支持 ROCm。
• JAX: 一个用于自动微分和数值计算的库，可以与 ROCm 集成。

以上是 TheRock 项目的最佳实践指南，希望能帮助您更好地理解和运用这个构建系统。