From 102392df5b71b71992f43701567747194f274daf Mon Sep 17 00:00:00 2001 From: YC Yen-Ching Tseng Date: Tue, 16 Jun 2026 16:13:22 +0800 Subject: [PATCH] [AMD][Fix] Skip EPLB topk remap when global server args are unset (#28404) --- python/sglang/srt/layers/moe/topk.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/python/sglang/srt/layers/moe/topk.py b/python/sglang/srt/layers/moe/topk.py index 10280df0c..3e9bb990a 100644 --- a/python/sglang/srt/layers/moe/topk.py +++ b/python/sglang/srt/layers/moe/topk.py @@ -1118,7 +1118,13 @@ def _eplb_remap_enabled() -> bool: # both unnecessary and not well-defined over the padded region of topk_ids). from sglang.srt.server_args import get_global_server_args - server_args = get_global_server_args() + try: + server_args = get_global_server_args() + except ValueError: + # Global server args are not initialized outside the server runtime + # (e.g. in unit tests that call select_experts directly). In that case + # there is no EPLB mapping, so the remap must be skipped. + return False return ( server_args.enable_eplb or server_args.init_expert_location != "trivial"