Update phoenix.mako template (#822)

Co-authored-by: Benjamin Alexander Wilfong <bwilfong3@login-phoenix-rh9-6.pace.gatech.edu>

Author: wilfonba

toolchain/templates/phoenix.mako

@@ -18,8 +18,7 @@
 #SBATCH --qos=${quality_of_service}
 % endif
 % if gpu:
-#SBATCH --gres=gpu:V100:${tasks_per_node}
-#SBATCH --mem-per-gpu=16G\
+#SBATCH -G${tasks_per_node}
 % endif
 % if email:
 #SBATCH --mail-user=${email}
@@ -41,10 +40,15 @@ echo
     % if not mpi:
         (set -x; ${profiler} "${target.get_install_binpath(case)}")
     % else:
-        (set -x; ${profiler}    \
-            mpirun -np ${nodes*tasks_per_node}            \
-                   --bind-to none                         \
-                   "${target.get_install_binpath(case)}")
+         (set -x; srun \
+            % if engine == 'interactive':
+                --nodes ${nodes} --ntasks-per-node ${tasks_per_node} \
+                --cpus-per-task 1                                    \
+                % if gpu:
+                    --gpus-per-task 1 --gpu-bind closest                 \
+                % endif
+            % endif
+            ${profiler} "${target.get_install_binpath(case)}")
     % endif
 
     ${helpers.run_epilogue(target)}