Enable uneven K#109

LucasWilkinson merged 2 commits intovllm-project:mainvllm-project/flash-attention:mainfrom

MatthewBonanni:uneven_kMatthewBonanni/flash-attention:uneven_kCopy head branch name to clipboard