From 2104d20eba2b3badcc51c3d8fbcec48f45e045e5 Mon Sep 17 00:00:00 2001
From: Atream <80757050+Atream@users.noreply.github.com>
Date: Thu, 6 Nov 2025 12:01:03 +0800
Subject: [PATCH] Temporarily fix missing routed_scaling_factor for
 CompressedTensorsWNA16MoEMethod (#12738)

---
 python/sglang/srt/models/deepseek_v2.py | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/python/sglang/srt/models/deepseek_v2.py b/python/sglang/srt/models/deepseek_v2.py
index cc33d6773..d3c364681 100644
--- a/python/sglang/srt/models/deepseek_v2.py
+++ b/python/sglang/srt/models/deepseek_v2.py
@@ -84,6 +84,7 @@ from sglang.srt.layers.quantization import CompressedTensorsConfig
 from sglang.srt.layers.quantization.base_config import QuantizationConfig
 from sglang.srt.layers.quantization.compressed_tensors.compressed_tensors_moe import (
     CompressedTensorsWNA16AMXEPMoEMethod,
+    CompressedTensorsWNA16MoEMethod,
 )
 from sglang.srt.layers.quantization.fp8 import Fp8Config
 from sglang.srt.layers.quantization.fp8_kernel import (
@@ -777,8 +778,14 @@ class DeepseekV2MoE(nn.Module):
             router_logits = self.gate(hidden_states, gemm_output_zero_allocator)
             topk_output = self.topk(hidden_states, router_logits)
             final_hidden_states = self.experts(hidden_states, topk_output)
-            if not _is_cuda or isinstance(
-                self.experts.quant_method, CompressedTensorsWNA16AMXEPMoEMethod
+            if (
+                not _is_cuda
+                or isinstance(
+                    self.experts.quant_method, CompressedTensorsWNA16AMXEPMoEMethod
+                )
+                or isinstance(
+                    self.experts.quant_method, CompressedTensorsWNA16MoEMethod
+                )
             ):
                 final_hidden_states *= self.routed_scaling_factor
 
@@ -838,7 +845,14 @@ class DeepseekV2MoE(nn.Module):
                 else {}
             ),
         )
-        if not _is_cuda and not _use_aiter:
+        if (
+            not _is_cuda
+            and not _use_aiter
+            or isinstance(
+                self.experts.quant_method, CompressedTensorsWNA16AMXEPMoEMethod
+            )
+            or isinstance(self.experts.quant_method, CompressedTensorsWNA16MoEMethod)
+        ):
             # fused in biased_grouped_topk so we can skip here
             final_hidden_states *= self.routed_scaling_factor
         if shared_output is not None: