diff --git a/psiflow/parsl_utils.py b/psiflow/parsl_utils.py index bcf0c87..a820c63 100644 --- 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(