From d067622820d459707cc94f217cf9555d3ed922bb Mon Sep 17 00:00:00 2001 From: Bingxu Chen Date: Tue, 25 Aug 2026 16:24:52 +0800 Subject: [PATCH] [AMD][bugfix] Add moe_ep_size/moe_tp_size to the allreduce-fusion gate test stub (#35340) --- test/registered/ops/test_aiter_allreduce_fusion_amd.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/test/registered/ops/test_aiter_allreduce_fusion_amd.py b/test/registered/ops/test_aiter_allreduce_fusion_amd.py index 69ed9a194..7b99f4b6c 100755 --- a/test/registered/ops/test_aiter_allreduce_fusion_amd.py +++ b/test/registered/ops/test_aiter_allreduce_fusion_amd.py @@ -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