[bugfix] Fix FilteredTopK overflow correctness#2605

yzh119 merged 1 commit intoflashinfer-ai:mainflashinfer-ai/flashinfer:mainfrom

jiangyinzuo:fix/bf16-topk-filtered-overflowjiangyinzuo/flashinfer:fix/bf16-topk-filtered-overflowCopy head branch name to clipboard