[CuTe,Fwd,SM90] Enable head dim 512 for SM90#2422

IwakuraRein wants to merge 2 commits into

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

IwakuraRein:enable-hdim-512-sm90IwakuraRein/flash-attention:enable-hdim-512-sm90Copy head branch name to clipboard