Skip to content

Commit 0877750

Browse files
authored
[CI/Build] Split pooling and generation extended language models tests in CI (#18705)
Signed-off-by: Isotr0py <[email protected]>
1 parent 6d68030 commit 0877750

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -486,16 +486,25 @@ steps:
486486
- pip freeze | grep -E 'torch'
487487
- pytest -v -s models/language -m core_model
488488

489-
- label: Language Models Test (Extended)
489+
- label: Language Models Test (Extended Generation) # 1hr20min
490490
mirror_hardwares: [amdexperimental]
491491
optional: true
492492
source_file_dependencies:
493493
- vllm/
494-
- tests/models/language
494+
- tests/models/language/generation
495495
commands:
496496
# Install causal-conv1d for plamo2 models here, as it is not compatible with pip-compile.
497497
- pip install 'git+https://github.com/Dao-AILab/[email protected]'
498-
- pytest -v -s models/language -m 'not core_model'
498+
- pytest -v -s models/language/generation -m 'not core_model'
499+
500+
- label: Language Models Test (Extended Pooling) # 36min
501+
mirror_hardwares: [amdexperimental]
502+
optional: true
503+
source_file_dependencies:
504+
- vllm/
505+
- tests/models/language/pooling
506+
commands:
507+
- pytest -v -s models/language/pooling -m 'not core_model'
499508

500509
- label: Multi-Modal Models Test (Standard)
501510
mirror_hardwares: [amdexperimental]

0 commit comments

Comments
 (0)