[FlexAttention] allow custom mask mod#37692

zou3519 merged 1 commit intovllm-project:mainvllm-project/vllm:mainfrom

liangel-02:flexliangel-02/vllm:flexCopy head branch name to clipboard