From 8078544c017f8fe5d6ec0c4318146d955c27d2d6 Mon Sep 17 00:00:00 2001
From: Khanak Bhargava <56127611+khanakbhargava@users.noreply.github.com>
Date: Fri, 8 Mar 2024 14:51:17 -0500
Subject: [PATCH] Update perlmutter.submit

---
 job_scripts/perlmutter/perlmutter.submit | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/job_scripts/perlmutter/perlmutter.submit b/job_scripts/perlmutter/perlmutter.submit
index 50df7f4..f43cace 100644
--- 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=$?