[feat] Faster topk algorithm#3009

Merged

kahyunnam merged 11 commits intoflashinfer-ai:mainfrom

Aalanli:topk_general

Apr 20, 2026

Commits on Apr 19, 2026

topk work

Allan Lin
authored and
Aalanli
committed
cleanup

Allan Lin
authored and
Aalanli
committed
fix merge
Aalanli
committed
adjust smem carveout for occupancy
Aalanli
committed
add feature gate
Aalanli
committed
refactor into topk_common.cuh
Aalanli
committed
precommit
Aalanli
committed
fix build errors on older sms
Aalanli
committed
address ai comments
Aalanli
committed
build fix for arch64
Aalanli
committed

Commits on Apr 20, 2026

fix bug
Aalanli
committed