minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
maocheng23
2026-03-18 21:57:05 +00:00
committed by GitHub
parent a3196d08b8
commit 4e8829e4cd
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/layers/moe/fused_moe_triton/fused_moe.py
+2 -2
View File
@@ -528,7 +528,7 @@ def fused_experts_impl(
intermediate_cache1.view(-1, N), intermediate_cache1.view(-1, N),
intermediate_cache2, intermediate_cache2,
config, config,
topk_ids, curr_topk_ids,
expert_ids, expert_ids,
down_moe_use_tma, down_moe_use_tma,
activation, activation,
@@ -554,7 +554,7 @@ def fused_experts_impl(
intermediate_cache1.view(-1, N), intermediate_cache1.view(-1, N),
intermediate_cache2, intermediate_cache2,
config, config,
topk_ids, curr_topk_ids,
expert_ids, expert_ids,
down_moe_use_tma, down_moe_use_tma,
activation, activation,