From 37a83007fef9925609a8d9b7c7b86bb41dab4e5d Mon Sep 17 00:00:00 2001 From: Andreas Karatzas Date: Thu, 26 Mar 2026 06:54:59 -0500 Subject: [PATCH] [ROCm][CI] Fix wvSplitKrc mock argument order in test_rocm_unquantized_gemm (#38167) Signed-off-by: Andreas Karatzas --- tests/model_executor/layers/test_rocm_unquantized_gemm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/model_executor/layers/test_rocm_unquantized_gemm.py b/tests/model_executor/layers/test_rocm_unquantized_gemm.py index e7cc0df7f..59d53dd60 100644 --- a/tests/model_executor/layers/test_rocm_unquantized_gemm.py +++ b/tests/model_executor/layers/test_rocm_unquantized_gemm.py @@ -76,7 +76,7 @@ def test_rocm_unquantized_gemm_gfx950_wvsplitkrc_path(monkeypatch): monkeypatch.setattr("vllm.platforms.rocm.on_gfx950", lambda: True) monkeypatch.setattr(utils, "num_compute_units", lambda: 120) - wvsplitkrc_mock = MagicMock(side_effect=lambda w, x_view, _, __: x_view @ w.t()) + wvsplitkrc_mock = MagicMock(side_effect=lambda x_view, w, _, __: x_view @ w.t()) monkeypatch.setattr(utils.ops, "wvSplitKrc", wvsplitkrc_mock) wvsplitk_mock = MagicMock(side_effect=lambda w, x_view, _, __: x_view @ w.t()) monkeypatch.setattr(utils.ops, "wvSplitK", wvsplitk_mock)