AMD ROCm 软件

ROCm 是一个主要由开源软件组成的开源软件栈，专为图形处理单元（GPU）计算而设计。ROCm 包含一系列驱动程序、开发工具和 API，支持从低级内核到终端用户应用程序的 GPU 编程。

使用 ROCm，您可以根据特定需求定制 GPU 软件。您可以在一个免费、开源、集成和安全的软件生态系统中开发、协作、测试和部署应用程序。ROCm 特别适合 GPU 加速的高性能计算（HPC）、人工智能（AI）、科学计算和计算机辅助设计（CAD）。

ROCm 由 AMD 的异构计算可移植接口（HIP）驱动，这是一个开源的 C++ GPU 编程环境及其相应的运行时。HIP 允许 ROCm 开发人员通过在不同平台上部署代码来创建可移植应用程序，从专用游戏 GPU 到百亿亿次级 HPC 集群。

ROCm 支持 OpenMP 和 OpenCL 等编程模型，并包含所有必要的开源软件编译器、调试器和库。ROCm 完全集成到机器学习（ML）框架中，如 PyTorch 和 TensorFlow。

获取 ROCm 源代码

AMD ROCm 由开源软件构建而成。因此，可以通过下载源代码并重新构建组件来修改 ROCm 的各个组件。可以使用 git 从每个 GitHub 仓库克隆 ROCm 组件的源代码。为了方便访问和下载这些工具的正确版本，ROCm 仓库包含一个名为 default.xml 的 repo 清单文件。您可以使用此清单文件下载 ROCm 软件的源代码。

安装 repo 工具

Google 的 repo 工具允许您同时管理多个 git 仓库。运行以下命令安装 repo 工具：

mkdir -p ~/bin/
curl https://storage.googleapis.com/git-repo-downloads/repo > ~/bin/repo
chmod a+x ~/bin/repo

注意： 这里使用 ~/bin/ 文件夹作为示例。如果需要，您可以指定不同的文件夹来安装 repo 工具。

安装 git-lfs

一些 ROCm 项目使用 Git 大文件存储（LFS）格式，可能需要您安装 git-lfs。更多信息请参考 Git 大文件存储。例如，要在 Ubuntu 上安装 git-lfs，请使用以下命令：

sudo apt-get install git-lfs

下载 ROCm 源代码

以下示例展示如何使用 repo 工具下载 ROCm 源代码。如果您选择安装 repo 工具的目录不是 ~/bin/，则必须在代码中使用您选择的目录，如下所示：

mkdir -p ~/ROCm/
cd ~/ROCm/
~/bin/repo init -u http://github.com/ROCm/ROCm.git -b roc-6.0.x
~/bin/repo sync

注意： 使用此示例代码将导致 repo 工具下载与指定 ROCm 版本相关的开源代码。在下载之前，请确保已按照 使用 SSH 连接到 GitHub 中的说明在您的机器上为 GitHub ID 配置了 ssh 密钥。

构建 ROCm 源代码

每个 ROCm 组件仓库都包含构建该组件的说明，例如 rocSPARSE 文档中的 Linux 安装和构建。有关构建仓库的说明，请参阅特定组件文档。

ROCm 软件的每个版本都支持特定的硬件和软件配置。有关当前支持的硬件和操作系统，请参阅 系统要求（Linux）。

从源代码构建 ROCm

构建过程将使用尽可能多的处理器进行并行构建。某些编译可能消耗高达 10GB 的 RAM，因此请确保有足够的交换空间！

默认情况下，ROCm 构建将针对所有支持的 GPU 架构进行编译，大约需要 500 个 CPU 小时。 如果我们使用环境变量 GPU_ARCHS 限制需要构建的 GPU 架构，构建时间将显著减少，如下所述。

# --------------------------------------
# 步骤1：克隆源代码
# --------------------------------------

mkdir -p ~/WORKSPACE/      # 或任何其他文件夹名称而不是 WORKSPACE
cd ~/WORKSPACE/
export ROCM_VERSION=6.1.0   # 或 6.1.1 6.1.2
~/bin/repo init -u http://github.com/ROCm/ROCm.git -b roc-6.1.x -m tools/rocm-build/rocm-${ROCM_VERSION}.xml
~/bin/repo sync

# --------------------------------------
# 步骤 2：准备构建环境
# --------------------------------------
# 选项1：启动 Docker 容器
# 拉取所需的基础 Docker 镜像：
# 从 ROCm/tools/rocm-build/docker/ubuntu20/Dockerfile 构建的 Ubuntu20.04
docker pull rocm/rocm-build-ubuntu-20.04:6.1
# 从 ROCm/tools/rocm-build/docker/ubuntu22/Dockerfile 构建的 Ubuntu22.04
docker pull rocm/rocm-build-ubuntu-22.04:6.1

# 启动 Docker 容器并挂载源代码文件夹：
docker run -ti \
    -e ROCM_VERSION=${ROCM_VERSION} \
    -e CCACHE_DIR=$HOME/.ccache \
    -e CCACHE_ENABLED=true \
    -e DOCK_WORK_FOLD=/src \
    -w /src \
    -v $PWD:/src \
    -v /etc/passwd:/etc/passwd \
    -v /etc/shadow:/etc/shadow \
    -v ${HOME}/.ccache:${HOME}/.ccache \
    -u $(id -u):$(id -g) \
    <替换为所需的 Ubuntu 基础 Docker 镜像> bash

# 选项2：在主机上安装所需的包
# 对于 Ubuntu20.04 系统
cd ROCm/tools/rocm-build/docker/ubuntu20
bash install-prerequisites.sh
# 对于 Ubuntu22.04 系统
cd ROCm/tools/rocm-build/docker/ubuntu22
bash install-prerequisities.sh

# --------------------------------------
# 步骤3：运行构建命令行
# --------------------------------------

# 在构建前选择 GPU 目标：
# 当未设置 GPU_ARCHS 时，将使用 ROCm6.1 支持的默认 GPU 目标。
# 要针对特定的 GFX 架构子集进行构建，可以使用以下环境变量。
# 支持 MI300（gfx940、gfx941、gfx942）。
export GPU_ARCHS="gfx942"               # 示例
export GPU_ARCHS="gfx940;gfx941;gfx942" # 示例

# 在 Docker 容器中选择并运行构建命令：
# 构建 rocm-dev 包
make -f ROCm/tools/rocm-build/ROCm.mk -j ${NPROC:-$(nproc)} rocm-dev
# 构建所有 ROCm 包
make -f ROCm/tools/rocm-build/ROCm.mk -j ${NPROC:-$(nproc)} all
# 列出所有 ROCm 组件以查找所需组件
make -f ROCm/tools/rocm-build/ROCm.mk list_components
# 构建单个 ROCm 包
make -f ROCm/tools/rocm-build/ROCm.mk T_rocblas

# 在 Ubuntu20.04 中找到构建的包：
out/ubuntu-20.04/20.04/deb/
# 在 Ubuntu22.04 中找到构建的包：
out/ubuntu-22.04/22.04/deb/

# 在 Ubuntu20.04 中找到构建日志：
out/ubuntu-20.04/20.04/logs/
# 在 Ubuntu22.04 中找到构建日志：
out/ubuntu-22.04/22.04/logs/
# 所有与失败组件相关的日志以 .errors 扩展名结尾。
out/ubuntu-22.04/22.04/logs/rocblas.errors          # 示例
# 所有与构建组件相关的日志以 .inprogress 扩展名结尾。
out/ubuntu-22.04/22.04/logs/rocblas.inprogress  # 示例
# 所有与通过组件相关的日志使用组件名称。
out/ubuntu-22.04/22.04/logs/rocblas             # 示例

注：ROCm.mk 概述

ROCm 文档

本仓库包含 ROCm 发布的清单文件、变更日志和发布信息。

default.xml 文件包含用于构建当前 ROCm 发布版本的所有仓库和相关提交信息；default.xml 使用清单格式仓库。

我们文档的源代码位于大多数 ROCm 仓库的 /docs 文件夹中。我们仓库的 develop 分支包含下一个 ROCm 发布版本的内容。

ROCm 文档主页是 rocm.docs.amd.com。

构建文档

快速开始构建，请使用以下代码。有关更多选项和详细信息，请参阅构建文档。

cd docs
pip3 install -r sphinx/requirements.txt
python3 -m sphinx -T -E -b html -d _build/doctrees -D language=en . _build/html

另外，还支持 CMake 构建。

cmake -B build
cmake --build build --target=doc

旧版 ROCm 发布

有关旧版 ROCm 发布的信息，请参阅变更日志。