From a0283fb5bbc94d9c488518eca9d98e0b6b2a8f47 Mon Sep 17 00:00:00 2001
From: youkaichao <youkaichao@gmail.com>
Date: Tue, 21 Jan 2025 21:47:16 +0800
Subject: [PATCH] Update test-template-aws.j2

---
 scripts/test-template-aws.j2 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/test-template-aws.j2 b/scripts/test-template-aws.j2
index 3c74ece..6ec8580 100644
--- 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"