Skip to content

Commit

Permalink
[PyTorch] Reduce the number of FA versions in L3 tests (#1280)
Browse files Browse the repository at this point in the history
remove one FA version in the L3 test

Signed-off-by: Charlene Yang <[email protected]>
  • Loading branch information
cyanguwa authored Oct 21, 2024
1 parent 3ea7dd3 commit 29e3a09
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion qa/L3_pytorch_FA_versions_test/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ pip install pytest==8.2.1
export MAX_JOBS=4

# Iterate over Flash Attention versions
FA_versions=(2.1.1 2.3.0 2.4.0.post1 2.4.1 2.5.7 2.6.3 3.0.0b1)
FA_versions=(2.1.1 2.3.0 2.4.1 2.5.7 2.6.3 3.0.0b1)
for fa_version in "${FA_versions[@]}"
do

Expand Down

0 comments on commit 29e3a09

Please sign in to comment.