[Cute] Fix two tests that were failing #2149

drisspg merged 2 commits intoDao-AILab:mainDao-AILab/flash-attention:mainfrom

henrylhtsang:fix_compute_capachenrylhtsang/flash-attention:fix_compute_capacCopy head branch name to clipboard