Prhub

#47065 [ROCm][CI] Move PyTorch Compilation Unit Tests to MI300(gfx942)

原始 PR 作者 charlifu 合并时间 2026-07-01 00:32 文件变更 1 提交数 3 评论 4 代码增减 +20 / -20

执行摘要

将 PyTorch 编译单元测试从 MI250 迁移到 MI300

为了让 PyTorch 编译单元测试在 MI300 硬件上运行,利用其更强的计算能力。PR 标题明确为 '[ROCm][CI] Move PyTorch Compilation Unit Tests to MI300(gfx942)'。

属于常规 CI 硬件迁移,实现简单,无精读必要。但可作为 CI 配置变更的参考案例。

讨论亮点

Review 中 AndreasKaratzas 指出了两次配置错误:

  1. 第一次提交中 mirror_hardwares 仍包含 amdgfx90anightlyamdmi250,未更新为 MI300 对应的值。
  2. 第二次提交后再次提醒 amdgfx90anightly 应改为 amdgfx942nightly
    提交者随后通过第三次提交修正了这些问题,最终获得批准。

实现拆解

  1. 移除 MI250 下的旧测试:从 MI250 (gfx90a) 测试区块中删除 PyTorch Compilation Unit Tests 的完整定义(共 18 行)。
  2. 在 MI300 区块新增测试:在 MI300 (gfx942) 测试区块中插入相同配置的测试,关键差异在于:
    • mirror_hardwares[amdexperimental, amdproduction, amdgfx90anightly, amdmi250] 改为 [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
    • agent_poolmi250_1 改为 mi300_1
  3. 修复 review 中发现的遗漏:第二次提交修正了首次提交中未正确更新的 mirror_hardwares 字段,将 amdgfx90anightly 改为 amdgfx942nightly
  4. 清理与对齐:确保其他相关硬件标签一致性,如将 amdmi250 改为 amdmi300
文件 模块 状态 重要度
.buildkite/test-amd.yaml CI 配置 modified 4.63

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

评论区精华

硬件标签未完全更新 正确性

AndreasKaratzas 指出 `mirror_hardwares` 中仍包含旧的 `amdgfx90anightly` 和 `amdmi250`,应改为 `amdgfx942nightly` 和 `amdmi300`。

结论:提交者后续提交修正了该问题。 · 已解决

风险与影响

低风险。变更仅涉及 CI 配置文件,无业务逻辑或运行时影响。可能的风险是 MI300 硬件资源不足导致测试排队或超时,但测试原为 optional: true,不会阻塞流水线。

直接影响 AMD CI 流水线:PyTorch 编译单元测试将从 MI250 迁移到 MI300 执行。MI300 性能更强,可能缩短测试时间,但测试的 timeout_in_minutes 仍为 180 分钟。对其他平台(NVIDIA、Intel 等)无影响。

仅 CI 配置变更

关联 Issue

未识别关联 Issue

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

完整报告

参与讨论