Add num_splits for mha_varlen_fwd FA2, support batch invariant#110

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

quanliu1991:mainquanliu1991/vllm-flash-attention:mainCopy head branch name to clipboard