fix broken test_mask_mod and test_score_mod#2152

brandon-yujie-sun wants to merge 1 commit intoDao-AILab:mainDao-AILab/flash-attention:mainfrom

brandon-yujie-sun:fix_broken_testsbrandon-yujie-sun/flash-attention:fix_broken_testsCopy head branch name to clipboard