Add pack-gqa fwd support for sparse impl w/ broadcasted H dim#2098

tridao merged 1 commit intomainDao-AILab/flash-attention:mainfrom

pack-gqa-backwardsDao-AILab/flash-attention:pack-gqa-backwardsCopy head branch name to clipboard