22 lines
460 B
YAML
22 lines
460 B
YAML
group: Samplers
|
|
depends_on:
|
|
- image-build
|
|
steps:
|
|
- label: Samplers Test
|
|
timeout_in_minutes: 75
|
|
source_file_dependencies:
|
|
- vllm/model_executor/layers
|
|
- vllm/sampling_metadata.py
|
|
- tests/samplers
|
|
- tests/conftest.py
|
|
commands:
|
|
- pytest -v -s samplers
|
|
- VLLM_USE_FLASHINFER_SAMPLER=1 pytest -v -s samplers
|
|
mirror:
|
|
amd:
|
|
device: mi325_1
|
|
depends_on:
|
|
- image-build-amd
|
|
commands:
|
|
- pytest -v -s -m samplers
|