Fix slurm mpipingpong job templates (#1874)

* add missing I_MPI_xxx values for slurm * add exclusive
Azure · Mar 5, 2024 · 2ede563 · 2ede563
1 parent 785757f
commit 2ede563
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/playbooks/files/ood_templates/slurm/impi_pingpong/pingpong.sh b/playbooks/files/ood_templates/slurm/impi_pingpong/pingpong.sh
@@ -5,6 +5,7 @@
 #SBATCH -p hpc
 #SBATCH -t 5
 #SBATCH --export=NONE
+#SBACTH --exclusive
 
 source /etc/profile.d/modules.sh
 module use /usr/share/Modules/modulefiles
@@ -13,7 +14,7 @@ module load mpi/impi-2021
 
 export I_MPI_DEBUG=5
 export I_MPI_HYDRA_BOOTSTRAP=slurm
-#export I_MPI_PORT_RANGE=59000:61000
+export I_MPI_PORT_RANGE=59000:61000
 
 echo "Running on $SLURM_NNODES nodes"
 mpirun -ppn 1 IMB-MPI1 PingPong
diff --git a/playbooks/files/ood_templates/slurm/impi_ring_pingpong/ring_pingpong.sh b/playbooks/files/ood_templates/slurm/impi_ring_pingpong/ring_pingpong.sh
@@ -5,6 +5,7 @@
 #SBATCH -p hpc
 #SBATCH -t 5
 #SBATCH --export=NONE
+#SBACTH --exclusive
 
 TYPE=latency
 
@@ -24,10 +25,9 @@ module use /usr/share/Modules/modulefiles
 
 module load mpi/impi-2021
 
-#export I_MPI_FABRICS="shm:ofi"
 export I_MPI_DEBUG=5
 export I_MPI_HYDRA_BOOTSTRAP=slurm
-#export FI_PROVIDER=mlx
+export I_MPI_PORT_RANGE=59000:61000
 
 IMB_ROOT=$I_MPI_ROOT/bin