Relax SM90 shared-memory estimate for wide values#16

ivandobskygithub merged 2 commits intomainivandobskygithub/flash-attention:mainfrom

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