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

add nvidia-smi to GH200 tests
vllm-project · Jan 21, 2025 · 8bcae2b · 8bcae2b
2 parents 7ca7ff4 + a0283fb
commit 8bcae2b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/test-template-aws.j2 b/scripts/test-template-aws.j2
@@ -267,7 +267,7 @@ steps:
     soft_fail: true
     agents:
       queue: gh200_queue
-    command: bash .buildkite/run-gh200-test.sh
+    command: nvidia-smi && bash .buildkite/run-gh200-test.sh
   {% endif %}
 
   - label: "TPU Test"