Update test-template-aws.j2

vllm-project · Jul 14, 2024 · 8cf1f82 · 8cf1f82
1 parent 8ad00e2
commit 8cf1f82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/test-template-aws.j2 b/scripts/test-template-aws.j2
@@ -127,7 +127,7 @@ steps:
           {% if step.label == "Benchmarks" %}
           mount-buildkite-agent: true
           {% endif %}
-          command: ["bash", "-c", "cd {{ (step.working_dir or default_working_dir) | safe  }} && {{ step.command  or (step.commands | join(' && ')) | safe }}"]
+          command: ["bash", "-c", "nvidia-smi && cd {{ (step.working_dir or default_working_dir) | safe  }} && {{ step.command  or (step.commands | join(' && ')) | safe }}"]
           environment:
             - VLLM_USAGE_SOURCE=ci-test
             - HF_HOME={{ hf_home }}