Skip to content

Commit 878fd5a

Browse files
authored
[CI/Build] Enable some fixed tests in AMD CI (#28078)
Signed-off-by: zhewenli <zhewenli@meta.com>
1 parent 18b3982 commit 878fd5a

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

.buildkite/test-amd.yaml

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ steps:
4848
commands:
4949
- bash standalone_tests/pytorch_nightly_dependency.sh
5050

51-
- label: Async Engine, Inputs, Utils, Worker Test # 36min
52-
timeout_in_minutes: 50
51+
- label: Async Engine, Inputs, Utils, Worker Test # 10min
52+
timeout_in_minutes: 15
5353
mirror_hardwares: [amdexperimental, amdproduction]
5454
agent_pool: mi325_1
5555
# grade: Blocking
@@ -616,9 +616,9 @@ steps:
616616
- uv pip install --system torchao==0.13.0
617617
- VLLM_TEST_FORCE_LOAD_FORMAT=auto pytest -v -s quantization/ --ignore quantization/test_blackwell_moe.py
618618

619-
- label: LM Eval Small Models # 53min
620-
timeout_in_minutes: 75
621-
mirror_hardwares: [amdexperimental]
619+
- label: LM Eval Small Models # 15min
620+
timeout_in_minutes: 20
621+
mirror_hardwares: [amdexperimental, amdproduction]
622622
agent_pool: mi325_1
623623
# grade: Blocking
624624
source_file_dependencies:
@@ -627,8 +627,8 @@ steps:
627627
commands:
628628
- pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=configs/models-small.txt --tp-size=1
629629

630-
- label: OpenAI API correctness # 22min
631-
timeout_in_minutes: 30
630+
- label: OpenAI API correctness # 10min
631+
timeout_in_minutes: 15
632632
mirror_hardwares: [amdexperimental, amdproduction]
633633
agent_pool: mi325_1
634634
# grade: Blocking
@@ -859,10 +859,10 @@ steps:
859859
- pytest -v -s models/multimodal -m core_model --ignore models/multimodal/generation/test_whisper.py --ignore models/multimodal/processing
860860
- cd .. && VLLM_WORKER_MULTIPROC_METHOD=spawn pytest -v -s tests/models/multimodal/generation/test_whisper.py -m core_model # Otherwise, mp_method="spawn" doesn't work
861861

862-
- label: Multi-Modal Accuracy Eval (Small Models) # 50min
863-
mirror_hardwares: [amdexperimental]
862+
- label: Multi-Modal Accuracy Eval (Small Models) # 10min
863+
mirror_hardwares: [amdexperimental, amdproduction]
864864
agent_pool: mi325_1
865-
timeout_in_minutes: 70
865+
timeout_in_minutes: 15
866866
working_dir: "/vllm-workspace/.buildkite/lm-eval-harness"
867867
source_file_dependencies:
868868
- vllm/multimodal/

0 commit comments

Comments
 (0)