From 692b8e5589975e645fef66d2f811ad3febc3f46b Mon Sep 17 00:00:00 2001
From: Spencer Bryngelson <sbryngelson@gmail.com>
Date: Mon, 29 Jan 2024 02:12:37 -0500
Subject: [PATCH] Update submit.sh

---
 .github/workflows/phoenix/submit.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/phoenix/submit.sh b/.github/workflows/phoenix/submit.sh
index d82d6aa5a..465c67cc4 100644
--- a/.github/workflows/phoenix/submit.sh
+++ b/.github/workflows/phoenix/submit.sh
@@ -15,11 +15,13 @@ fi
 
 sbatch_cpu_opts="\
 #SBATCH -p cpu-small               # partition
+#SBATCH -N1                        # Number of nodes required
 #SBATCH --ntasks-per-node=24       # Number of cores per node required
 #SBATCH --mem-per-cpu=2G           # Memory per core\
 "
 
 sbatch_gpu_opts="\
+#SBATCH -n4                        # Number of ranks required
 #SBATCH -CV100-16GB
 #SBATCH -G2\
 "
@@ -39,7 +41,6 @@ sbatch <<EOT
 #!/bin/bash
 #SBATCH -Jshb-$job_slug            # Job name
 #SBATCH --account=gts-sbryngelson3 # charge account
-#SBATCH -N1                        # Number of nodes required
 $sbatch_device_opts
 #SBATCH -t 04:00:00                # Duration of the job (Ex: 15 mins)
 #SBATCH -q embers                  # QOS Name