perf: optimize top-k kernel for fp32#2915

Archie-wang wants to merge 2 commits intoflashinfer-ai:mainflashinfer-ai/flashinfer:mainfrom

Archie-wang:optimize-topk-fp32Archie-wang/flashinfer:optimize-topk-fp32Copy head branch name to clipboard