From ec0f632911d06fc00d65527683b7217fec1aba66 Mon Sep 17 00:00:00 2001
From: youkaichao <youkaichao@126.com>
Date: Tue, 9 Jul 2024 22:00:08 -0700
Subject: [PATCH] add workdir and command list

---
 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 35ea92d..3293f45 100644
--- 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 %}