ensure parsl launch scripts do not get saved in cwd

molmod · Feb 11, 2024 · 129bf87 · 129bf87
1 parent 066c499
commit 129bf87
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/psiflow/parsl_utils.py b/psiflow/parsl_utils.py
@@ -80,12 +80,14 @@ def __call__(self, command: str, tasks_per_node: int, nodes_per_block: int) -> s
 [[ "{debug}" == "1" ]] && echo "Found nodes : $NODES"
 WORKERCOUNT={task_blocks}
 
-cat << SLURM_EOF > cmd_$SLURM_JOB_NAME.sh
+path_cmd=$(dirname $SLURM_JOB_STDOUT)
+
+cat << SLURM_EOF > $path_cmd/cmd_$SLURM_JOB_NAME.sh
 {command}
 SLURM_EOF
-chmod a+x cmd_$SLURM_JOB_NAME.sh
+chmod a+x $path_cmd/cmd_$SLURM_JOB_NAME.sh
 
-srun --ntasks {task_blocks} -l {overrides} bash cmd_$SLURM_JOB_NAME.sh
+srun --ntasks {task_blocks} -l {overrides} bash $path_cmd/cmd_$SLURM_JOB_NAME.sh
 
 [[ "{debug}" == "1" ]] && echo "Done"
 """.format(