[EPLB] Drop defensive getattr for ep_dispatch_algorithm (#31804)

Co-authored-by: Brayden Zhong <b8zhong@uwaterloo.ca>
This commit is contained in:
Beihao Zhou
2026-09-15 16:14:41 +08:00
committed by GitHub
co-authored by Brayden Zhong
parent df254b0a11
commit 37ebacb50f
3 changed files with 8 additions and 6 deletions
@@ -64,7 +64,9 @@ class TestWaterfillEPLB(CustomTestCase):
num_fused_shared_experts=1,
routed_scaling_factor=1.0,
)
dispatch_info = SimpleNamespace(num_physical_experts=264)
dispatch_info = SimpleNamespace(
num_physical_experts=264, ep_dispatch_algorithm="static"
)
def fake_eplb_postprocess(
ids, expert_location_dispatch_info, num_token_non_padded
@@ -104,7 +106,9 @@ class TestWaterfillEPLB(CustomTestCase):
num_fused_shared_experts=1,
routed_scaling_factor=1.0,
)
dispatch_info = SimpleNamespace(num_physical_experts=264)
dispatch_info = SimpleNamespace(
num_physical_experts=264, ep_dispatch_algorithm="static"
)
def fake_eplb_postprocess(
ids, expert_location_dispatch_info, num_token_non_padded