Merge pull request #12 from youkaichao/ykc-multi-node

add workdir and command list
vllm-project · Jul 10, 2024 · fc8fa9a · fc8fa9a
2 parents 73d2203 + ec0f632
commit fc8fa9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/test-template-aws.j2 b/scripts/test-template-aws.j2
@@ -67,7 +67,7 @@ steps:
     agents:
       queue: gpu_4_queue
     commands:
-      - ./.buildkite/run-multi-node-test.sh {{ step.num_nodes }} {{ step.num_gpus }} {{ docker_image }} {% for command in step.commands %}"{{ command }}" {% endfor %}
+      - ./.buildkite/run-multi-node-test.sh {{ (step.working_dir or default_working_dir) | safe  }} {{ step.num_nodes }} {{ step.num_gpus }} {{ docker_image }} {% for command in step.commands %}"{{ (command | join(" && ")) | safe }}" {% endfor %}
   {% endif %}
   {% endfor %}