Merge pull request #26 from khanakbhargava/patch-1

Update perlmutter.submit
AMReX-Astro · Mar 9, 2024 · 73ec0eb · 73ec0eb
2 parents 87036cb + 8078544
commit 73ec0eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/job_scripts/perlmutter/perlmutter.submit b/job_scripts/perlmutter/perlmutter.submit
@@ -76,7 +76,7 @@ slack_job_start.py "starting NERSC job: ${workdir} ${restartFile}" @michael
 
 # execute srun in the background then use the builtin wait so the shell can
 # handle the signal
-srun -n 64 ${CASTRO_EXEC} ${INPUTS} ${restartString} &
+srun -n $((SLURM_NTASKS_PER_NODE * SLURM_NNODES)) ${CASTRO_EXEC} ${INPUTS} ${restartString} &
 pid=$!
 wait $pid
 ret=$?