[Misc] Add num_splits input arg to flash_attn_varlen_func#72

WoosukKwon merged 1 commit intomainvllm-project/flash-attention:mainfrom

woosuk/add-num-splitsvllm-project/flash-attention:woosuk/add-num-splitsCopy head branch name to clipboard