Clamp SM90 tiles to GMMA-friendly sizes#15

ivandobskygithub merged 2 commits into

mainivandobskygithub/flash-attention:mainfrom

codex/fix-flashattention-shared-memory-error-i7s50livandobskygithub/flash-attention:codex/fix-flashattention-shared-memory-error-i7s50lCopy head branch name to clipboard