add mpi bootstrap (#1862)

Azure · Feb 27, 2024 · 1068a4f · 1068a4f
1 parent d656566
commit 1068a4f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/playbooks/files/ood_templates/slurm/impi_pingpong/pingpong.sh b/playbooks/files/ood_templates/slurm/impi_pingpong/pingpong.sh
@@ -11,5 +11,9 @@ module use /usr/share/Modules/modulefiles
 
 module load mpi/impi-2021
 
+export I_MPI_DEBUG=5
+export I_MPI_HYDRA_BOOTSTRAP=slurm
+#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
@@ -26,6 +26,7 @@ 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
 
 IMB_ROOT=$I_MPI_ROOT/bin