执行摘要

• 一句话：升级 ROCm Dockerfile 中 UCX 和 RIXL 版本
• 推荐动作：值得合并，修复 CI 问题。可直接合入。

功能与动机

根据 PR body 描述，升级 UCX 和 RIXL 以解决 CI 构建失败问题，引用 Buildkite 构建链接。

实现拆解

1. 在 docker/Dockerfile.rocm 中，将 RIXL_BRANCH 从 bf4a7214 更新为 39be1de8，将 UCX_BRANCH 从 7009d7a1 更新为 bfb51733。
2. 在 RIXL 构建步骤中，添加环境变量 _ucx_install_dir=${UCX_HOME}，以确保构建脚本能正确找到 UCX 安装目录。

关键文件：

• docker/Dockerfile.rocm（模块 部署脚本；类别 infra；类型 infrastructure）: 唯一变更文件，升级 UCX 和 RIXL 版本并修复构建环境变量。

关键符号：未识别

关键源码片段

docker/Dockerfile.rocm

唯一变更文件，升级 UCX 和 RIXL 版本并修复构建环境变量。

# RIXL/UCX build stages
FROM base AS build_rixl
​
# 升级 RIXL 到新 commit 以修复 CI 构建问题
ARG RIXL_BRANCH="39be1de8"
ARG RIXL_REPO="https://github.com/ROCm/RIXL.git"
​
# 升级 UCX 到新 commit
ARG UCX_BRANCH="bfb51733"
ARG UCX_REPO="https://github.com/openucx/ucx.git"
​
ENV ROCM_PATH=/opt/rocm
ENV UCX_HOME=/usr/local/ucx
​
RUN cd /opt/rixl && \
    sed -i "s/--exclude 'libamdhip64*'/--exclude 'libamdhip64*' --exclude 'libcore*' --exclude 'libpull*'/" \
    contrib/build-wheel.sh && \
    mkdir -p /app/install && \
    # 显式设置 UCX 安装目录，确保 RIXL 构建能正确找到 UCX
    _ucx_install_dir=${UCX_HOME} \
    ./contrib/build-wheel.sh \
      --output-dir /app/install \
      --rocm-dir ${ROCM_PATH} \

评论区精华

无实质性讨论。gemini-code-assist[bot] 确认变更内容，tjtanaa 批准。

• 暂无高价值评论线程

风险与影响

• 风险：风险低。仅是依赖版本更新和环境变量添加，不涉及核心逻辑变更。但新的依赖版本可能引入兼容性问题，需通过 CI 验证。
• 影响：影响仅限 ROCm 平台的 Docker 构建流程。修复 CI 构建失败，对用户无直接影响。
• 风险标记：依赖版本更新

关联脉络

• PR #37646 [ROCm][FEAT] AITER Fused Allreduce + RMSNorm: 同为 ROCm 平台基础设施变更，涉及 CI 构建依赖。
• PR #41217 [ROCm][Deepseek] dsv3.2 further optimization: 同为 ROCm 平台相关 PR，该 PR 的 CI 构建可能依赖此处升级的 UCX/RIXL。