From 24d42f3553a35aa53006a3ee8d04cc9191cb7be9 Mon Sep 17 00:00:00 2001
From: "Kevin H. Luu" <khluu000@gmail.com>
Date: Mon, 31 Aug 2026 14:56:27 -0700
Subject: [PATCH] [CI] Mark 1-GPU L4 test steps with device: l4 for EKS
 migration (#54549)

Signed-off-by: khluu <khluu000@gmail.com>
---
 .buildkite/test_areas/distributed.yaml     | 1 +
 .buildkite/test_areas/engine.yaml          | 2 ++
 .buildkite/test_areas/kernels.yaml         | 6 ++++++
 .buildkite/test_areas/models_language.yaml | 2 ++
 .buildkite/test_areas/pytorch.yaml         | 4 ++++
 5 files changed, 15 insertions(+)

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