minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Bingxu Chen
2026-08-25 01:24:52 -07:00
committed by GitHub
parent 191244b3f6
commit d067622820
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/ops/test_aiter_allreduce_fusion_amd.py
+5 -1
View File
@@ -409,11 +409,15 @@ class TestAiterAllreduceFusionGate(CustomTestCase):
) )
) )
stack.enter_context(mock.patch.object(comm, "_use_aiter", use_aiter)) 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( stack.enter_context(
mock.patch.object( mock.patch.object(
comm, comm,
"get_parallel", "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 # the gate reads get_exec().comm.enable_aiter_allreduce_fusion