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

[Test] Fix fusion-group mocks after runtime context migration (#40290)

Browse Source
This commit is contained in:
Xiaoyu Zhang
2026-09-19 14:45:16 +08:00
committed by GitHub
parent 5d703de9e4
commit 36aa8479ef
1 changed files with 11 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/unit/layers/test_layer_communicator_fusion_gate.py
+11 -22
View File
@@ -169,28 +169,17 @@ class TestResolveFusionGroup(CustomTestCase):
fake_moe_tp_group = MagicMock(name="moe_tp_group")
tp_size = moe_ep_size * moe_tp_size * moe_dp_size
with (
get_parallel().override(
moe_ep_size=moe_ep_size,
moe_tp_size=moe_tp_size,
moe_dp_size=moe_dp_size,
tp_size=tp_size,
tp_rank=tp_rank,
moe_ep_rank=tp_rank % moe_ep_size,
moe_tp_rank=tp_rank % moe_tp_size,
),
patch(
"sglang.srt.layers.flashinfer_comm_fusion.get_tp_group",
return_value=fake_tp_group,
),
patch(
"sglang.srt.layers.flashinfer_comm_fusion.get_moe_ep_group",
return_value=fake_ep_group,
),
patch(
"sglang.srt.layers.flashinfer_comm_fusion.get_moe_tp_group",
return_value=fake_moe_tp_group,
),
with get_parallel().override(
moe_ep_size=moe_ep_size,
moe_tp_size=moe_tp_size,
moe_dp_size=moe_dp_size,
tp_size=tp_size,
tp_rank=tp_rank,
moe_ep_rank=tp_rank % moe_ep_size,
moe_tp_rank=tp_rank % moe_tp_size,
tp_group=fake_tp_group,
moe_ep_group=fake_ep_group,
moe_tp_group=fake_moe_tp_group,
):
ws = resolve_fusion_world_size(use_attn_tp_group=False)
group_tuple = resolve_fusion_group(use_attn_tp_group=False)