[sgl-kernel] update deepgemm (#6942)

2025-06-07 14:24:41 +08:00
parent c4ffbeca19
commit 22fe787852
2 changed files with 5 additions and 2 deletions
--- a/python/sglang/srt/models/deepseek_v2.py
+++ b/python/sglang/srt/models/deepseek_v2.py
@@ -1719,7 +1719,10 @@ class DeepseekV2ForCausalLM(nn.Module):
                    logger,
                    "Only Deepseek V3/R1 on NV-platform can use shared experts fusion optimization. Shared experts fusion optimization is disabled.",
                )
-            elif (global_server_args_dict["enable_deepep_moe"] or global_server_args_dict["enable_ep_moe"]):
+            elif (
+                global_server_args_dict["enable_deepep_moe"]
+                or global_server_args_dict["enable_ep_moe"]
+            ):
                self.num_fused_shared_experts = 0
                global_server_args_dict["disable_shared_experts_fusion"] = True
                log_info_on_rank0(