From 1068a4fa1da96d3a37092490774c1e0936eb70ab Mon Sep 17 00:00:00 2001 From: Xavier Pillons Date: Tue, 27 Feb 2024 10:58:56 +0100 Subject: [PATCH] add mpi bootstrap (#1862) --- playbooks/files/ood_templates/slurm/impi_pingpong/pingpong.sh | 4 ++++ .../ood_templates/slurm/impi_ring_pingpong/ring_pingpong.sh | 1 + 2 files changed, 5 insertions(+) diff --git a/playbooks/files/ood_templates/slurm/impi_pingpong/pingpong.sh b/playbooks/files/ood_templates/slurm/impi_pingpong/pingpong.sh index 4ed490d76..e0d7f88c3 100644 --- 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 index ab786fd4b..c9a653ca8 100644 --- 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