[BUG FIX] Optimize top-k mask construction: prevent unsafe gradient flow and eliminate unnecessary memory allocations#184

LoserCheems wants to merge 1 commit intomainflash-algo/flash-sparse-attention:mainfrom

fix-180flash-algo/flash-sparse-attention:fix-180Copy head branch name to clipboard