Update test-template-aws.j2

vllm-project · Dec 16, 2024 · 6a8e18c · 6a8e18c
1 parent db48f69
commit 6a8e18c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/test-template-aws.j2 b/scripts/test-template-aws.j2
@@ -255,14 +255,14 @@ steps:
     agents:
       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: