Use tile size 128 x 96 for hdim 64,256#116

LucasWilkinson merged 1 commit intovllm-project:mainvllm-project/flash-attention:mainfrom

PatrykSaffer:patryk/cherry-pick-shape-fixPatrykSaffer/flash-attention:patryk/cherry-pick-shape-fixCopy head branch name to clipboard