Isolate Q TMA setup behind tag dispatch#19

ivandobskygithub merged 2 commits intomainivandobskygithub/flash-attention:mainfrom

codex/revert-csrc-changes-and-validate-hopper-i67d3bivandobskygithub/flash-attention:codex/revert-csrc-changes-and-validate-hopper-i67d3bCopy head branch name to clipboard