diff --git a/.buildkite/test_areas/distributed.yaml b/.buildkite/test_areas/distributed.yaml index 461a47d8de..8825a67b1b 100644 --- a/.buildkite/test_areas/distributed.yaml +++ b/.buildkite/test_areas/distributed.yaml @@ -20,6 +20,7 @@ steps: - label: ":nvidia: (L4) Sharded RDT Weight Transfer" key: sharded-rdt-weight-transfer + device: l4 timeout_in_minutes: 20 working_dir: "/vllm-workspace/tests" num_devices: 1 diff --git a/.buildkite/test_areas/engine.yaml b/.buildkite/test_areas/engine.yaml index 32b1d60690..4384484c09 100644 --- a/.buildkite/test_areas/engine.yaml +++ b/.buildkite/test_areas/engine.yaml @@ -38,6 +38,8 @@ steps: - label: ":nvidia: (L4) V1 Engine" key: engine-1-gpu + device: l4 + num_devices: 1 timeout_in_minutes: 45 source_file_dependencies: - vllm/v1/engine/ diff --git a/.buildkite/test_areas/kernels.yaml b/.buildkite/test_areas/kernels.yaml index 8ce562d0d0..fc13887799 100644 --- a/.buildkite/test_areas/kernels.yaml +++ b/.buildkite/test_areas/kernels.yaml @@ -104,6 +104,8 @@ steps: - label: ":nvidia: (L4) Attention Kernels Shard %N" key: kernels-attention-test + device: l4 + num_devices: 1 timeout_in_minutes: 65 source_file_dependencies: - csrc/attention/ @@ -165,6 +167,8 @@ steps: - label: ":nvidia: (L4) Quantization Kernels Shard %N" key: kernels-quantization-test + device: l4 + num_devices: 1 timeout_in_minutes: 60 source_file_dependencies: - csrc/quantization/ @@ -194,6 +198,8 @@ steps: - label: ":nvidia: (L4) MoE Kernels Shard %N" key: kernels-moe-test + device: l4 + num_devices: 1 timeout_in_minutes: 50 source_file_dependencies: - csrc/quantization/cutlass_w8a8/moe/ diff --git a/.buildkite/test_areas/models_language.yaml b/.buildkite/test_areas/models_language.yaml index c55f733fd9..f6fa53832e 100644 --- a/.buildkite/test_areas/models_language.yaml +++ b/.buildkite/test_areas/models_language.yaml @@ -92,6 +92,8 @@ steps: # correctness test on L4 until its H200 output matches the Transformers reference. - label: ":nvidia: (L4) Granite Language Model Compatibility" key: language-models-tests-granite-l4-compatibility + device: l4 + num_devices: 1 timeout_in_minutes: 65 source_file_dependencies: - vllm/ diff --git a/.buildkite/test_areas/pytorch.yaml b/.buildkite/test_areas/pytorch.yaml index a5e39f7b66..eb5b840703 100644 --- a/.buildkite/test_areas/pytorch.yaml +++ b/.buildkite/test_areas/pytorch.yaml @@ -90,6 +90,8 @@ steps: - label: ":nvidia: (L4) PyTorch Compilation Passes" key: pytorch-compilation-passes-unit-tests + device: l4 + num_devices: 1 timeout_in_minutes: 45 source_file_dependencies: - vllm/__init__.py @@ -164,6 +166,8 @@ steps: # those SM90 cases skip while the architecture-compatible cases still run. - label: ":nvidia: (L4) PyTorch Fullgraph CUDAGraph Compatibility" key: pytorch-fullgraph-cudagraph-l4-compatibility + device: l4 + num_devices: 1 timeout_in_minutes: 60 source_file_dependencies: - vllm/__init__.py