Metal backend: Add SDPA head_dim=256 support#18875

manuelcandales merged 5 commits intomainpytorch/executorch:mainfrom

gh/manuelcandales/170/headpytorch/executorch:gh/manuelcandales/170/headCopy head branch name to clipboard