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,