[CI] Fix MXFP8 MOE backend selection tests on gfx942 (#50222)
Signed-off-by: Felix Marty <Felix.Marty@amd.com>
fxmarty-amd committed
Jul 29, 2026 at 11:17 UTC
5b14019576475224d86044b262e28a04a85d4086
1 file changed
+11
-7
tests/kernels/moe/test_mxfp8_aiter_backend_selection.py
+11
-7
@@ -27,8 +27,8 @@ from vllm.model_executor.layers.fused_moe.experts.aiter_mxfp8_moe import ( # no
27
_AITER_SWIGLU_BETA,
28
AiterMxfp8Experts,
29
)
30
-from vllm.model_executor.layers.fused_moe.experts.mxfp8_native_moe import ( # noqa: E402
31
- Mxfp8NativeTritonExperts,
30
+from vllm.model_executor.layers.fused_moe.experts.mxfp8_emulation_moe import ( # noqa: E402
31
+ Mxfp8EmulationTritonExperts,
32
)
33
from vllm.model_executor.layers.fused_moe.modular_kernel import ( # noqa: E402
34
FusedMoEActivationFormat,
@@ -150,16 +150,20 @@ def test_explicit_moe_backend_aiter():
150
151
def test_gfx950_picks_aiter():
152
"""Auto-select on real ROCm hardware with flydsl usable -> FlyDSL wins."""
153
- with _flydsl_installed(True):
153
+ # NOTE: Fp8MoeBackend.AITER_MXFP8 does not require VLLM_ROCM_USE_AITER=1
154
+ with (
155
+ patch(f"{_AITER_MOD}.current_platform.supports_mx", return_value=True),
156
+ _flydsl_installed(True),
157
+ ):
158
backend, experts_cls = select_mxfp8_moe_backend(_config())
159
assert backend is Fp8MoeBackend.AITER_MXFP8
160
assert experts_cls is AiterMxfp8Experts
161
162
159
-def test_gfx942_picks_triton():
163
+def test_gfx942_picks_emulation():
164
"""flydsl unusable (e.g. gfx942, no FlyDSL support) -> native Triton
165
dot_scaled backend wins instead."""
162
- with _flydsl_installed(False):
166
+ with patch(f"{_AITER_MOD}.current_platform.supports_mx", return_value=False):
167
backend, experts_cls = select_mxfp8_moe_backend(_config())
164
- assert backend is Fp8MoeBackend.TRITON_MXFP8
165
- assert experts_cls is Mxfp8NativeTritonExperts
168
+ assert backend is Fp8MoeBackend.EMULATION
169
+ assert experts_cls is Mxfp8EmulationTritonExperts