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
+4 -15
View File
@@ -169,8 +169,7 @@ class TestResolveFusionGroup(CustomTestCase):
fake_moe_tp_group = MagicMock(name="moe_tp_group") fake_moe_tp_group = MagicMock(name="moe_tp_group")
tp_size = moe_ep_size * moe_tp_size * moe_dp_size tp_size = moe_ep_size * moe_tp_size * moe_dp_size
with ( with get_parallel().override(
get_parallel().override(
moe_ep_size=moe_ep_size, moe_ep_size=moe_ep_size,
moe_tp_size=moe_tp_size, moe_tp_size=moe_tp_size,
moe_dp_size=moe_dp_size, moe_dp_size=moe_dp_size,
@@ -178,19 +177,9 @@ class TestResolveFusionGroup(CustomTestCase):
tp_rank=tp_rank, tp_rank=tp_rank,
moe_ep_rank=tp_rank % moe_ep_size, moe_ep_rank=tp_rank % moe_ep_size,
moe_tp_rank=tp_rank % moe_tp_size, moe_tp_rank=tp_rank % moe_tp_size,
), tp_group=fake_tp_group,
patch( moe_ep_group=fake_ep_group,
"sglang.srt.layers.flashinfer_comm_fusion.get_tp_group", moe_tp_group=fake_moe_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,
),
): ):
ws = resolve_fusion_world_size(use_attn_tp_group=False) ws = resolve_fusion_world_size(use_attn_tp_group=False)
group_tuple = resolve_fusion_group(use_attn_tp_group=False) group_tuple = resolve_fusion_group(use_attn_tp_group=False)