[Bugfix][ROCm][MoE] Update AITER MXFP4 W4A16 tests to the renamed expert_mask (#51980)
Signed-off-by: Stefan Koncarevic <Stefan.Koncarevic@amd.com>
stefankoncarevic committed
Aug 12, 2026 at 19:45 UTC
d20a031d33fb8d56cfe090681872319544040da7
1 file changed
+2
-2
tests/kernels/moe/test_rocm_aiter_moe.py
+2
-2
@@ -651,7 +651,7 @@ def test_aiter_fused_moe_mi350_mxfp4_w4a16_accuracy():
651
activation=MoEActivation.SWIGLUOAI,
652
quant_config=case.quant_config,
653
moe_config=case.moe_config,
654
- expert_map=None,
654
+ expert_mask=None,
655
)
656
657
assert out.shape == case.hidden_states.shape
@@ -695,7 +695,7 @@ def test_aiter_fused_moe_mi350_mxfp4_w4a16_determinism():
695
activation=MoEActivation.SWIGLUOAI,
696
quant_config=case.quant_config,
697
moe_config=case.moe_config,
698
- expert_map=None,
698
+ expert_mask=None,
699
)
700
701
_assert_deterministic(run_mxfp4_moe, n_runs=4)