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: