From 29e3a09052cfbe4d8971df7ed436700cdfc26726 Mon Sep 17 00:00:00 2001 From: Charlene Yang <8636796+cyanguwa@users.noreply.github.com> Date: Mon, 21 Oct 2024 14:43:08 -0700 Subject: [PATCH] [PyTorch] Reduce the number of FA versions in L3 tests (#1280) remove one FA version in the L3 test Signed-off-by: Charlene Yang <8636796+cyanguwa@users.noreply.github.com> --- qa/L3_pytorch_FA_versions_test/test.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qa/L3_pytorch_FA_versions_test/test.sh b/qa/L3_pytorch_FA_versions_test/test.sh index 162ed85823..6c23e39a48 100644 --- a/qa/L3_pytorch_FA_versions_test/test.sh +++ b/qa/L3_pytorch_FA_versions_test/test.sh @@ -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