[misc] use out argument for flash attention#9740

youkaichao wants to merge 4 commits intovllm-project:mainvllm-project/vllm:mainfrom

youkaichao:flash_outyoukaichao/vllm:flash_outCopy head branch name to clipboard