feat: implement max_logits support for flashmask#139

xxyux wants to merge 1 commit intoPaddlePaddle:mainPaddlePaddle/flash-attention:mainfrom

xxyux:max_logitsxxyux/flash-attention:max_logitsCopy head branch name to clipboard