FA3 tensor size parameter fix for long context len (seqlen >=4M)#1841

tridao merged 1 commit into

Dao-AILab:mainDao-AILab/flash-attention:mainfrom

ghadiaravi13:mainghadiaravi13/flash-attention:mainCopy head branch name to clipboard