From 792cb3a5d069d3ce7b94a4c337a1cab039ebaa6d Mon Sep 17 00:00:00 2001 From: Lawrence Wu Date: Thu, 18 Jun 2026 12:51:09 -0700 Subject: [PATCH] fix: add missing guard for use_jit_ep_activation (#27377) --- python/sglang/srt/layers/moe/moe_runner/deep_gemm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/layers/moe/moe_runner/deep_gemm.py b/python/sglang/srt/layers/moe/moe_runner/deep_gemm.py index a814b5d23..4c3fda886 100644 --- a/python/sglang/srt/layers/moe/moe_runner/deep_gemm.py +++ b/python/sglang/srt/layers/moe/moe_runner/deep_gemm.py @@ -872,7 +872,7 @@ def _varlen_deep_gemm_silu_mul_quant( ) use_jit_ep_activation = envs.SGLANG_OPT_USE_JIT_EP_ACTIVATION.get() - if N % 4 != 0 or G % 4 != 0: + if N % 4 != 0 or G % 4 != 0 or D // 8 < E: use_jit_ep_activation = False if use_jit_ep_activation: