Replace topk_ids with curr_topk_ids in fused_moe.py (#20302)

2026-03-18 14:57:05 -07:00
parent a3196d08b8
commit 4e8829e4cd
1 changed files with 2 additions and 2 deletions
--- 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,