From 0d3bf6b5c21451a93b0f962bbe57387a941ad476 Mon Sep 17 00:00:00 2001 From: LuminolT Date: Mon, 6 Jul 2026 14:13:55 +0800 Subject: [PATCH] fix(moe): import capture mode from cuda graph runner Tested: PYTHONPYCACHEPREFIX=/private/tmp/sglang_pycache python3 -m py_compile python/sglang/srt/layers/moe/mega_moe.py test/registered/unit/moe/test_glm_megamoe.py test/registered/unit/server_args/test_server_args.py --- python/sglang/srt/layers/moe/mega_moe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/layers/moe/mega_moe.py b/python/sglang/srt/layers/moe/mega_moe.py index 5313e4002..4510cf683 100644 --- a/python/sglang/srt/layers/moe/mega_moe.py +++ b/python/sglang/srt/layers/moe/mega_moe.py @@ -26,7 +26,7 @@ from sglang.srt.eplb.expert_location_dispatch import ExpertLocationDispatchInfo from sglang.srt.layers.dp_attention import get_dp_global_num_tokens from sglang.srt.layers.moe.topk import select_experts from sglang.srt.layers.moe.utils import get_moe_a2a_backend -from sglang.srt.model_executor.runner import get_is_capture_mode +from sglang.srt.model_executor.cuda_graph_runner import get_is_capture_mode if TYPE_CHECKING: from deep_gemm import SymmBuffer