diff --git a/apps/imb-mpi/allreduce.sh b/apps/imb-mpi/allreduce.sh index 59b2b45a..5cd59629 100755 --- a/apps/imb-mpi/allreduce.sh +++ b/apps/imb-mpi/allreduce.sh @@ -9,21 +9,30 @@ num_ranks=$(wc -l <$PBS_NODEFILE) case $MPI in impi2018) module load mpi/impi - source $MPI_BIN/mpivars.sh + #source $MPI_BIN/mpivars.sh export I_MPI_FABRICS="shm:ofa" export I_MPI_FALLBACK_DEVICE=0 export I_MPI_DEBUG=4 mpi_options="-hostfile $PBS_NODEFILE -np $num_ranks" - IMB_ROOT=$MPI_BIN + if [ -z $MPI_BIN ]; then + IMB_ROOT=$I_MPI_ROOT/intel64/bin + else + IMB_ROOT=$MPI_BIN + fi ;; impi2019) module load mpi/impi-2019 - source $MPI_BIN/mpivars.sh -ofi_internal + #source $MPI_BIN/mpivars.sh -ofi_internal export I_MPI_FABRICS="shm:ofi" export I_MPI_FALLBACK_DEVICE=0 export I_MPI_DEBUG=4 + export FI_PROVIDER=mlx mpi_options="-hostfile $PBS_NODEFILE -np $num_ranks" - IMB_ROOT=$MPI_BIN + if [ -z $MPI_BIN ]; then + IMB_ROOT=$I_MPI_ROOT/intel64/bin + else + IMB_ROOT=$MPI_BIN + fi ;; ompi) module load mpi/hpcx diff --git a/apps/imb-mpi/ringpingpong.sh b/apps/imb-mpi/ringpingpong.sh index 01b5d8f2..0c1b32a2 100755 --- a/apps/imb-mpi/ringpingpong.sh +++ b/apps/imb-mpi/ringpingpong.sh @@ -7,23 +7,32 @@ module use /usr/share/Modules/modulefiles case $MPI in impi2018) module load mpi/impi - source $MPI_BIN/mpivars.sh + #source $MPI_BIN/mpivars.sh export I_MPI_FABRICS="shm:ofa" export I_MPI_FALLBACK_DEVICE=0 export I_MPI_DEBUG=4 mpi_options="-np 2 -ppn 1" host_option="-hosts" - IMB_ROOT=$MPI_BIN + if [ -z $MPI_BIN ]; then + IMB_ROOT=$I_MPI_ROOT/intel64/bin + else + IMB_ROOT=$MPI_BIN + fi ;; impi2019) module load mpi/impi-2019 - source $MPI_BIN/mpivars.sh -ofi_internal + #source $MPI_BIN/mpivars.sh -ofi_internal export I_MPI_FABRICS="shm:ofi" export I_MPI_FALLBACK_DEVICE=0 export I_MPI_DEBUG=4 + export FI_PROVIDER=mlx mpi_options="-np 2 -ppn 1" host_option="-hosts" - IMB_ROOT=$MPI_BIN + if [ -z $MPI_BIN ]; then + IMB_ROOT=$I_MPI_ROOT/intel64/bin + else + IMB_ROOT=$MPI_BIN + fi ;; ompi) module load mpi/hpcx