Update test-template-fastcheck.j2

vllm-project · Jul 14, 2024 · ac926dc · ac926dc
1 parent 8cf1f82
commit ac926dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/test-template-fastcheck.j2 b/scripts/test-template-fastcheck.j2
@@ -133,7 +133,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 }}