From 4e8829e4cd98b501f983ef3c3e758fa710476f4e Mon Sep 17 00:00:00 2001
From: maocheng23 <35615230+maocheng23@users.noreply.github.com>
Date: Wed, 18 Mar 2026 14:57:05 -0700
Subject: [PATCH] Replace topk_ids with curr_topk_ids in fused_moe.py (#20302)

---
 python/sglang/srt/layers/moe/fused_moe_triton/fused_moe.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/sglang/srt/layers/moe/fused_moe_triton/fused_moe.py b/python/sglang/srt/layers/moe/fused_moe_triton/fused_moe.py
index 75f62c955..f2621473c 100644
--- a/python/sglang/srt/layers/moe/fused_moe_triton/fused_moe.py
+++ b/python/sglang/srt/layers/moe/fused_moe_triton/fused_moe.py
@@ -528,7 +528,7 @@ def fused_experts_impl(
                         intermediate_cache1.view(-1, N),
                         intermediate_cache2,
                         config,
-                        topk_ids,
+                        curr_topk_ids,
                         expert_ids,
                         down_moe_use_tma,
                         activation,
@@ -554,7 +554,7 @@ def fused_experts_impl(
                         intermediate_cache1.view(-1, N),
                         intermediate_cache2,
                         config,
-                        topk_ids,
+                        curr_topk_ids,
                         expert_ids,
                         down_moe_use_tma,
                         activation,