[AMD][bugfix] Add moe_ep_size/moe_tp_size to the allreduce-fusion gate test stub (#35340)

2026-08-25 01:24:52 -07:00
parent 191244b3f6
commit d067622820
1 changed files with 5 additions and 1 deletions
@@ -409,11 +409,15 @@ class TestAiterAllreduceFusionGate(CustomTestCase):
                )
            )
            stack.enter_context(mock.patch.object(comm, "_use_aiter", use_aiter))
+            # moe_ep_size/moe_tp_size of 1 keep the hybrid EP+TP guard inactive
+            # so the aiter branch is what decides.
            stack.enter_context(
                mock.patch.object(
                    comm,
                    "get_parallel",
-                    lambda: types.SimpleNamespace(tp_size=tp_world_size),
+                    lambda: types.SimpleNamespace(
+                        tp_size=tp_world_size, moe_ep_size=1, moe_tp_size=1
+                    ),
                )
            )
            # the gate reads get_exec().comm.enable_aiter_allreduce_fusion