Fix out kwarg shape check with ngroups swapped#4

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

Yard1:fix_out_with_ngroups_swappedYard1/vllm-flash-attention:fix_out_with_ngroups_swappedCopy head branch name to clipboard