执行摘要

该 PR 为 AMD ROCm Dockerfile 添加了 AITER_COMMIT 构建参数支持，允许用户在构建时通过 --build-arg 覆盖 AITER 版本，同时保持默认值不变。变更涉及 docker/rocm.Dockerfile 和配套 CI 脚本，风险较低，主要提升构建灵活性和开发者体验。

功能与动机

此前，AITER 的 commit/tag 在 rocm.Dockerfile 的每个基础镜像阶段都是硬编码的（通过 ENV AITER_COMMIT），这意味着切换不同 AITER 版本需要直接编辑 Dockerfile，无法像其他依赖（如 TRITON_COMMIT、SGL_BRANCH）那样在构建时通过 --build-arg 参数化。该变更旨在提供构建时覆盖 AITER 版本的能力，便于测试和自定义构建，同时保持向后兼容。

实现拆解

1. docker/rocm.Dockerfile

• 重命名默认变量：将四个基础镜像阶段（gfx942、gfx942-rocm720、gfx950、gfx950-rocm720）中的 ENV AITER_COMMIT="v0.1.11.post1" 统一重命名为 ENV AITER_COMMIT_DEFAULT="v0.1.11.post1"。
• 新增构建参数：在公共参数区域添加：
  

  ARG AITER_COMMIT=""
  ENV AITER_COMMIT="${AITER_COMMIT:-${AITER_COMMIT_DEFAULT}}"
  

  
  这允许用户通过 --build-arg AITER_COMMIT=<ref> 覆盖版本，未指定时回退到默认值。

2. scripts/ci/amd/amd_ci_install_dependency.sh

• 更新提取逻辑：将 grep 和 sed 模式从匹配 AITER_COMMIT= 改为 AITER_COMMIT_DEFAULT=，确保 CI 脚本能正确提取默认版本信息，维持 CI 流程正常。

评论区精华

review 中主要讨论点：

gemini-code-assist[bot] 指出："The AITER_COMMIT_DEFAULT version is duplicated across four different stages... Consider defining a single global ARG at the top of the file and referencing it in each stage to centralize version management."
gemini-code-assist[bot] 建议："The extraction logic for AITER_COMMIT_DEFAULT is fragile... Using a more flexible regex with sed can make this more robust."
HaiShaw 补充："build-arg AITER_COMMIT will not be used to build official images"

这些讨论揭示了代码冗余和脚本健壮性问题，但未在本次 PR 中强制解决，为后续优化留出空间。

风险与影响

• 风险：
  • 维护风险：AITER_COMMIT_DEFAULT 在四个阶段重复定义，未来更新版本需多处修改，易引入不一致。
  • CI 脚本健壮性：版本提取逻辑依赖特定字符串格式，若 Dockerfile 格式化变更可能导致失败。
  • 构建风险：用户传递无效构建参数可能使用非预期版本，但影响限于自定义构建。
• 影响：
  • 对开发者：提供了构建时覆盖 AITER 版本的灵活性，简化测试流程。
  • 对系统：仅影响构建配置，不改变运行时行为或性能。
  • 对团队：小幅提升基础设施可配置性，但重复定义问题可能增加维护成本。

关联脉络

• 与 PR #21511 和 #21947 同属 AMD 相关改进，均涉及 ROCm 环境或 AITER 组件，反映了对 AMD 硬件支持持续优化的趋势。
• 与 PR #21447 类似，都是基础设施层面的依赖版本管理变更，体现了项目对构建灵活性和可维护性的关注。
• 从近期历史 PR 看，AMD 标签频繁出现，表明该仓库正积极投入 AMD 平台适配和性能优化，本 PR 是这一方向上的基础设施增强。