From effa414568838b3d64be290c762c197bcc0b7a45 Mon Sep 17 00:00:00 2001
From: Erik Strand <erik.strand@cba.mit.edu>
Date: Sat, 6 Mar 2021 15:17:11 -0500
Subject: [PATCH] Run on four GPUs (one node)

---
 mpi_pi_gpu/mpi_pi_gpu.slurm | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/mpi_pi_gpu/mpi_pi_gpu.slurm b/mpi_pi_gpu/mpi_pi_gpu.slurm
index f31083a..b9405af 100644
--- a/mpi_pi_gpu/mpi_pi_gpu.slurm
+++ b/mpi_pi_gpu/mpi_pi_gpu.slurm
@@ -3,13 +3,14 @@
 #SBATCH -o output/%j.out
 #SBATCH -e output/%j.err
 #SBATCH --nodes=1
-#SBATCH --ntasks-per-node=2
-#SBATCH --gres=gpu:2
+#SBATCH --ntasks-per-node=4
+#SBATCH --gres=gpu:4
+#SBATCH --exclusive
 #SBATCH --cpus-per-task=1
 #SBATCH --ntasks-per-core=1
 #SBATCH --threads-per-core=1
 #SBATCH --mem=10G
-#SBATCH --time 00:01:00
+#SBATCH --time 00:05:00
 
 source ./load_modules.sh
 srun ./mpi_pi_gpu
-- 
GitLab