viable/strict/1787658955: [CI] Add gfx950 FlyDSL Inductor test shard (#193473)
PyTorch 在 CI 中为 gfx950 添加了 FlyDSL Inductor 测试分片,安装 FlyDSL 0.3.0 轮子,并在 ROCm 运行时或设备不可用时提前失败,运行完整 FlyDSL 模板/GEMM 测试文件,本地测试 26 通过,7 子测试通过。
发展脉络
- 首次出现viable/strict/1787658955: [CI] Add gfx950 FlyDSL Inductor test shard (#193473)PyTorch Core
- 行业反馈trunk/78d28664274451624de1ebb10eaa8187663c4efd: [CI] Add gfx950 FlyDSL Inductor test shard (#193473)PyTorch Core
- 当前判断PyTorch 对 AMD 最新 GPU 的持续集成支持,反映了开源深度学习框架对多硬件供应商的重视。随着 AMD 在 AI 加速器市场的份额增长,框架层面的及时支持可能影响开发者的硬件选择。Agent Pulse · 分析
PyTorch 合并了 PR #193473,为 gfx950 添加了专用的 FlyDSL Inductor 测试分片。该变更安装固定的 FlyDSL 0.3.0 发布轮子,并在 trunk gfx950 矩阵中增加专用分片,当 ROCm 运行时或 gfx950 设备不可用时提前失败,而不是静默跳过 GEMM 覆盖。测试计划包括 shell 检查、lintrunner 和本地 gfx950 预检,运行 inductor/test_flydsl_template.py,26 个测试通过,7 个子测试通过。该 PR 由 jgong5 和 jansel 批准,依赖 #190903。
该变更表明 PyTorch Inductor 正在为 AMD gfx950 硬件增加专门的测试覆盖,使用 FlyDSL 作为模板/GEMM 的 DSL。提前失败机制确保 CI 不会在硬件不可用时静默跳过测试,提高了测试可靠性。这暗示 FlyDSL 在 gfx950 上的支持正在成熟,但具体性能或功能细节未在证据中提及。
PyTorch 对 AMD 最新 GPU 的持续集成支持,反映了开源深度学习框架对多硬件供应商的重视。随着 AMD 在 AI 加速器市场的份额增长,框架层面的及时支持可能影响开发者的硬件选择。
对于使用 AMD GPU 进行 AI 训练或推理的团队,此变更意味着 PyTorch 对 gfx950 的支持将更可靠,减少因硬件不兼容导致的 CI 失败,可能降低部署风险。
后续可关注 FlyDSL 在 gfx950 上的性能基准或更多测试覆盖的扩展,以及 PyTorch 对 AMD 其他 GPU 型号的支持情况。