Skip to content

Commit

Permalink
Update test-pipeline.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
khluu authored Nov 19, 2024
1 parent b00b33d commit 8402917
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions .buildkite/test-pipeline.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,7 @@
# label(str): the name of the test. emoji allowed.
# fast_check(bool): whether to run this on each commit on fastcheck pipeline.
# fast_check_only(bool): run this test on fastcheck pipeline only
# nightly(bool): run this test in nightly pipeline only
# optional(bool): never run this test by default (i.e. need to unblock manually)
# optional(bool): never run this test by default (i.e. need to unblock manually) unless it's scheduled nightly run.
# command(str): the single command to run for tests. incompatible with commands.
# commands(list): the list of commands to run for test. incompatbile with command.
# mirror_hardwares(list): the list of hardwares to run the test on as well. currently only supports [amd]
Expand Down Expand Up @@ -336,7 +335,7 @@ steps:
- pytest -v -s models/embedding/vision_language -m core_model

- label: Language Models Test (Extended) # 50min
nightly: true
optional: true
source_file_dependencies:
- vllm/
- tests/models/decoder_only/language
Expand All @@ -362,7 +361,7 @@ steps:
- pytest -v -s models/encoder_decoder/vision_language -m core_model

- label: Multi-Modal Models Test (Extended) # 1h15m
nightly: true
optional: true
source_file_dependencies:
- vllm/
- tests/models/decoder_only/audio_language
Expand Down Expand Up @@ -513,6 +512,7 @@ steps:

- label: Distributed Tests (A100) # optional
gpu: a100
optional: true
num_gpus: 4
source_file_dependencies:
- vllm/
Expand All @@ -526,6 +526,7 @@ steps:

- label: LM Eval Large Models # optional
gpu: a100
optional: true
num_gpus: 4
working_dir: "/vllm-workspace/.buildkite/lm-eval-harness"
source_file_dependencies:
Expand Down

0 comments on commit 8402917

Please sign in to comment.