vllm-project/vllm · 作者 AndreasKaratzas · 合并时间 2026-03-26 19:55
根据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 |
分析完成后，这里会展示 LLM 生成的相对完整源码片段和详细注释。
gemini-code-assist[bot] 确认修复无误，tjtanaa 批准合并。
结论：修复被接受并合并，无争议。 · 已解决
风险较低：仅修改测试代码，不影响生产逻辑。但需注意mock参数顺序错误可能导致测试假通过，从而隐藏真实问题；本次修复已纠正此问题。此外，无回归风险，因为变更单一且目标明确。
对最终用户无直接影响。对开发团队：修复了CI测试失败，提升了代码质量和CI可靠性，确保ROCm平台相关测试正确运行。影响范围限于ROCm平台的特定测试模块。
当前没有检测到明确关联的 Issue 链接，后续同步到相关引用后会出现在这里。
修复了ROCm平台中一个测试函数的mock参数顺序错误，解决了因复制粘贴错误导致的CI失败，确保测试套件稳定运行。
本PR旨在修复test_rocm_unquantized_gemm_gfx950_wvsplitkrc_path测试的失败问题。根据PR body描述，失败原因是mock设置中的复制粘贴错误：wvSplitKrc mock函数参数顺序与C++实现不匹配（wvSplitK和wvSplitKrc有相反的参数约定），导致返回张量形状错误，使torch.allclose断言失败。
变更仅涉及一个文件：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()。utils.py中ops.wvSplitKrc(x, weight, ...)的调用方式（activations first），确保mock返回正确的张量形状（16x256而非256×16）。review讨论简洁，主要包括：
风险分析：
影响评估：
从近期历史PR分析看，本PR是ROCm平台测试维护的一部分，与以下PR关联：
参与讨论