Prhub

#38167 [ROCm][CI] Fix wvSplitKrc mock argument order in test_rocm_unquantized_gemm

原始 PR 作者 AndreasKaratzas 合并时间 2026-03-26 19:55 文件变更 1 提交数 1 评论 1 代码增减 +1 / -1

执行摘要

修复 ROCm 测试中 wvSplitKrc mock 的参数顺序错误,确保 CI 通过。

根据PR body,修复了test_rocm_unquantized_gemm_gfx950_wvsplitkrc_path测试失败,原因是mock设置中的复制粘贴错误导致参数顺序颠倒(wvSplitKrc mock错误地复制了wvSplitK的参数顺序),返回错误形状的张量,使torch.allclose失败。

该PR变更简单,无需精读。值得快速查看以了解测试mock的使用方式和参数顺序敏感性的处理,但无复杂设计决策。

讨论亮点

review讨论简洁:gemini-code-assist[bot]确认修复无误,tjtanaa批准并合并。未出现技术争议或深入讨论,表明修复直接明了。

实现拆解

仅修改一个文件:tests/model_executor/layers/test_rocm_unquantized_gemm.py。在wvsplitkrc_mock的lambda函数中,将参数顺序从lambda w, x_view, _, __: x_view @ w.t()改为lambda x_view, w, _, __: x_view @ w.t(),以正确模拟wvSplitKrc函数的调用方式(activations first)。

文件 模块 状态 重要度
tests/model_executor/layers/test_rocm_unquantized_gemm.py tests/model_executor/layers modified 2.0

关键符号

wvsplitkrc_mock

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

评论区精华

修复 mock 参数顺序确认 测试

gemini-code-assist[bot] 确认修复无误,tjtanaa 批准合并。

结论:修复被接受并合并,无争议。 · 已解决

风险与影响

风险较低:仅修改测试代码,不影响生产逻辑。但需注意mock参数顺序错误可能导致测试假通过,从而隐藏真实问题;本次修复已纠正此问题。此外,无回归风险,因为变更单一且目标明确。

对最终用户无直接影响。对开发团队:修复了CI测试失败,提升了代码质量和CI可靠性,确保ROCm平台相关测试正确运行。影响范围限于ROCm平台的特定测试模块。

测试 mock 错误 参数顺序敏感

关联 Issue

未识别关联 Issue

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

完整报告

参与讨论