Prhub

#29219 [AMD-miles] Add a ROCm 7.0 MI35x miles nightly image

原始 PR 作者 JessicaJiang-123 合并时间 2026-06-25 11:27 文件变更 1 提交数 1 评论 1 代码增减 +4 / -1

执行摘要

为 AMD MI35x 添加 ROCm 7.0 夜间 Docker 构建

AMD MI35x 需要 ROCm 7.0 基础镜像:ROCm 7.2 存在已知的 VMM 物理内存释放回归问题,导致 torch_memory_saver 无法释放 VRAM,从而在 colocate sleep/wake 场景下 OOM。使用 ROCm 7.0 可避免该问题。

该 PR 是常规基础设施变更,无技术难度,但为解决 MI35x 实际问题的必要操作。值得关注的是其依赖的基础镜像 v0.5.10-rocm700-mi35x 的维护状况。

讨论亮点

未发现实质性的设计讨论。该 PR 获得 1 个 approvals(bingxche),无 review 评论。

实现拆解

  1. .github/workflows/release-docker-amd-miles-rocm700-nightly.yml 的构建矩阵中增加 gfx950 / rocm700-mi35x 条目,指定 rocm_tagrocm700-mi35x,基础镜像 tag 为 v0.5.10-rocm700-mi35x
  2. push_local_registry 矩阵中将 gpu_arch["gfx942"] 扩展为 ["gfx950", "gfx942"],确保 MI35x 镜像也能被推送到本地 registry。
  3. 变更后,ROCm 7.0 夜间流水线会同时构建 rocm/sgl-dev:miles-rocm700-mi35x-<DATE>(gfx950 / MI350X·MI355X)和 rocm/sgl-dev:miles-rocm700-mi30x-<DATE>(gfx942 / MI300)两个镜像。
文件 模块 状态 重要度
.github/workflows/release-docker-amd-miles-rocm700-nightly.yml CI modified 3.19

分析完成后,这里会展示 LLM 生成的相对完整源码片段和详细注释。

评论区精华

没有提炼出高价值讨论线程

当前评论区没有形成足够清晰的争议点或结论,后续有更多讨论时会体现在这里。

风险与影响

低风险。变更仅涉及 CI 配置,未触及任何源代码逻辑。若基础镜像 v0.5.10-rocm700-mi35x 不存在或构建失败,仅影响该矩阵条目对应的 nightly 构建,不影响其他构建。

对用户:AMD MI35x(gfx950)用户可获得专门的 ROCm 7.0 夜间 Docker 镜像,避免 ROCm 7.2 的 VMM 回归问题。
对系统/团队:新增一个 CI 构建条目,增加少量构建时间,但无功能影响。

依赖基础镜像 v0.5.10-rocm700-mi35x 是否存在

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论