Fix unpadding input with padding mask compute error#38

wwbitejotunn wants to merge 2 commits intoPaddlePaddle:mainPaddlePaddle/flash-attention:mainfrom

wwbitejotunn:varlen_mask_mainwwbitejotunn/flash-attention:varlen_mask_mainCopy head branch name to clipboard