[CI] Mark 1-GPU L4 test steps with device: l4 for EKS migration (#54549)

Signed-off-by: khluu <khluu000@gmail.com>
This commit is contained in:
Kevin H. Luu
2026-08-31 14:56:27 -07:00
committed by GitHub
parent f5c3cc240b
commit 24d42f3553
5 changed files with 15 additions and 0 deletions
+1
View File
@@ -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
+2
View File
@@ -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/
+6
View File
@@ -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/
@@ -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/
+4
View File
@@ -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