Add GH200 to fast check as an optional step

vllm-project · Jan 29, 2025 · 95bf0ca · 95bf0ca
1 parent 5f23bbd
commit 95bf0ca
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/scripts/test-template-fastcheck.j2 b/scripts/test-template-fastcheck.j2
@@ -216,3 +216,14 @@ steps:
     commands: 
       - if [[ -f ".buildkite/run-tpu-test.sh" ]]; then bash .buildkite/run-tpu-test.sh; fi 
       - yes | docker system prune -a
+
+  - block: "Run GH200 Test"
+    depends_on: ~
+    key: block-gh200
+
+  - label: "GH200 Test"
+    depends_on: block-gh200
+    soft_fail: true
+    agents:
+      queue: gh200_queue
+    command: nvidia-smi && bash .buildkite/run-gh200-test.sh