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"