Skip to content

Commit bb2135e

Browse files
AndreasKaratzasgeodavic
authored andcommitted
VLLM_USE_TRITON_FLASH_ATTN V0 variable deprecation (vllm-project#27611)
Signed-off-by: Andreas Karatzas <akaratza@amd.com> Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com> Signed-off-by: George D. Torres <gdavtor@gmail.com>
1 parent 7835398 commit bb2135e

File tree

15 files changed

+11
-1587
lines changed

15 files changed

+11
-1587
lines changed

.buildkite/scripts/hardware_ci/run-amd-test.sh

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -78,17 +78,13 @@ HF_MOUNT="/root/.cache/huggingface"
7878
commands=$@
7979
echo "Commands:$commands"
8080

81-
if [[ $commands == *"pytest -v -s basic_correctness/test_basic_correctness.py"* ]]; then
82-
commands=${commands//"pytest -v -s basic_correctness/test_basic_correctness.py"/"VLLM_USE_TRITON_FLASH_ATTN=0 pytest -v -s basic_correctness/test_basic_correctness.py"}
83-
fi
81+
commands=${commands//"pytest -v -s basic_correctness/test_basic_correctness.py"/"pytest -v -s basic_correctness/test_basic_correctness.py"}
8482

8583
if [[ $commands == *"pytest -v -s models/test_registry.py"* ]]; then
8684
commands=${commands//"pytest -v -s models/test_registry.py"/"pytest -v -s models/test_registry.py -k 'not BambaForCausalLM and not GritLM and not Mamba2ForCausalLM and not Zamba2ForCausalLM'"}
8785
fi
8886

89-
if [[ $commands == *"pytest -v -s compile/test_basic_correctness.py"* ]]; then
90-
commands=${commands//"pytest -v -s compile/test_basic_correctness.py"/"VLLM_USE_TRITON_FLASH_ATTN=0 pytest -v -s compile/test_basic_correctness.py"}
91-
fi
87+
commands=${commands//"pytest -v -s compile/test_basic_correctness.py"/"pytest -v -s compile/test_basic_correctness.py"}
9288

9389
if [[ $commands == *"pytest -v -s lora"* ]]; then
9490
commands=${commands//"pytest -v -s lora"/"VLLM_ROCM_CUSTOM_PAGED_ATTN=0 pytest -v -s lora"}

0 commit comments

Comments
 (0)