diff --git a/aiter/fused_moe.py b/aiter/fused_moe.py index 41771a8389..3f8cc65288 100644 --- a/aiter/fused_moe.py +++ b/aiter/fused_moe.py @@ -836,9 +836,10 @@ def _fused_moe_impl( and stage1_func in (_flydsl_stage1_wrapper, cktile_moe_stage1) and expert_mask is not None ) - assert ( - not metadata.flat or get_gfx() == "gfx950" - ), f"FLAT fmoe asm kernels are gfx950-only; refusing to launch on {get_gfx()}. " + assert not metadata.flat or get_gfx() in ( + "gfx942", + "gfx950", + ), f"FLAT fmoe asm kernels require gfx942/gfx950; got {get_gfx()}. " sort_m_indices = None sort_reverse_sorted = None