fix: topK uint32 overflow#2937

Linda-Stadter wants to merge 2 commits intoflashinfer-ai:mainflashinfer-ai/flashinfer:mainfrom

Linda-Stadter:topk_overflowLinda-Stadter/flashinfer:topk_overflowCopy head branch name to clipboard