Skip to content

Commit

Permalink
Merge pull request #201 from pyiron/slurm_cwd
Browse files Browse the repository at this point in the history
Fix working directory in SLURM
  • Loading branch information
jan-janssen authored Nov 7, 2023
2 parents a5f0142 + 1aef6b0 commit 69350ea
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion pympipool/shared/interface.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,9 @@ def generate_mpiexec_command(cores, oversubscribe=False):
def generate_slurm_command(
cores, cwd, threads_per_core=1, gpus_per_core=0, oversubscribe=False
):
command_prepend_lst = ["srun", "-n", str(cores), "-D", cwd]
command_prepend_lst = ["srun", "-n", str(cores)]
if cwd is not None:
command_prepend_lst += ["-D", cwd]
if threads_per_core > 1:
command_prepend_lst += ["--cpus-per-task" + str(threads_per_core)]
if gpus_per_core > 0:
Expand Down

0 comments on commit 69350ea

Please sign in to comment.