Re-introduce the unit test of test_mooncake_ep_small (#16019)

2026-01-12 17:01:24 +08:00
parent b1ee75ae7b
commit 9f5cd80a8d
4 changed files with 17 additions and 21 deletions
--- a/python/sglang/srt/eplb/eplb_algorithms/init.py
+++ b/python/sglang/srt/eplb/eplb_algorithms/init.py
@@ -54,7 +54,7 @@ def rebalance_experts(
            num_groups=num_groups,
            num_nodes=num_nodes,
            num_gpus=num_physical_experts // num_local_physical_experts,
-            enable_hierarchical=True,
+            enable_hierarchical=False,
            active_ranks=(
                ElasticEPStateManager.instance().active_ranks
                if ElasticEPStateManager.instance() is not None