From 169fc1e20cbf5c87c4f44e2bf1873249cb719d30 Mon Sep 17 00:00:00 2001
From: McZyWu <zhuoyun.wu.23@ucl.ac.uk>
Date: Mon, 27 Jul 2026 15:40:49 +0800
Subject: [PATCH] [NPU] Acc fix for afmoe model introduced by topk refactor.
 (#31280)

---
 python/sglang/srt/models/afmoe.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/python/sglang/srt/models/afmoe.py b/python/sglang/srt/models/afmoe.py
index 72bc43a6a..d9573e817 100644
--- a/python/sglang/srt/models/afmoe.py
+++ b/python/sglang/srt/models/afmoe.py
@@ -233,9 +233,7 @@ class AfmoeMoE(nn.Module):
                 expert_bias=self.expert_bias,
             )
 
-        renormalize = (
-            self.route_norm if self.score_func == "sigmoid" and not _is_npu else False
-        )
+        renormalize = self.route_norm if self.score_func == "sigmoid" else False
         self.topk = TopK(
             top_k=self.top_k,
             renormalize=renormalize,
@@ -245,7 +243,14 @@ class AfmoeMoE(nn.Module):
             custom_routing_function=custom_routing_fn,
             correction_bias=correction_bias,
             routed_scaling_factor=self.route_scale,
-            **({"scoring_func": self.score_func} if _is_npu else {}),
+            **(
+                {
+                    "scoring_func": self.score_func,
+                    "apply_routed_scaling_factor_on_output": True,
+                }
+                if _is_npu
+                else {}
+            ),
         )
 
     def pack_params(self) -> None: