Merge pull request #54 from vllm-project/youkaichao-patch-1

add gh200 tests
vllm-project · Dec 16, 2024 · 00d71d1 · 00d71d1
2 parents eff4cc4 + 770bf8a
commit 00d71d1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/scripts/test-template-aws.j2 b/scripts/test-template-aws.j2
@@ -256,6 +256,15 @@ steps:
       queue: ibm-ppc64le
     command: bash .buildkite/run-cpu-test-ppc64le.sh
 
+  {% if branch == "main" %}
+  - label: "GH200 Test"
+    depends_on: ~
+    soft_fail: true
+    agents:
+      queue: gh200_queue
+    command: bash .buildkite/run-gh200-test.sh
+  {% else %}
+
   - label: "TPU Test"
     depends_on: ~
     agents: