diff --git a/python/sglang/srt/models/deepseek_v2.py b/python/sglang/srt/models/deepseek_v2.py index 9aafb6edd..45a84a6fb 100644 --- a/python/sglang/srt/models/deepseek_v2.py +++ b/python/sglang/srt/models/deepseek_v2.py @@ -787,6 +787,7 @@ class DeepseekV2MoE(nn.Module): ModelOptFp4LinearMethod, ) and fc1_n % 128 == 0 + and self.shared_experts.swiglu_limit is None and not check_cuda_graph_backend(Phase.PREFILL, Backend.TC_PIECEWISE) ): self.shared_experts.gate_up_proj._interleave_for_swiglu_fusion = True