61 lines
1.6 KiB
Bash
61 lines
1.6 KiB
Bash
preCheckHooks+=('setupMpiCheck')
|
|
preInstallCheckHooks+=('setupMpiCheck')
|
|
|
|
|
|
setupMpiCheck() {
|
|
# Find out which MPI implementation we are using
|
|
# and set safe defaults that are guaranteed to run
|
|
# on any build machine
|
|
|
|
mpiType="NONE"
|
|
|
|
# OpenMPI signature
|
|
if command ompi_info &> /dev/null; then
|
|
mpiType="openmpi"
|
|
fi
|
|
|
|
# MPICH based implementations
|
|
if command mpichversion &> /dev/null; then
|
|
if [ "$mpiType" != "NONE" ]; then
|
|
echo "WARNING: found OpenMPI and MPICH/MVAPICH executables"
|
|
fi
|
|
|
|
version=$(mpichversion)
|
|
if [[ "$version" == *"MPICH"* ]]; then
|
|
mpiType="MPICH"
|
|
fi
|
|
if [[ "$version" == *"MVAPICH"* ]]; then
|
|
mpiType="MVAPICH"
|
|
fi
|
|
fi
|
|
|
|
echo "Found MPI implementation: $mpiType"
|
|
|
|
case $mpiType in
|
|
openmpi)
|
|
# Note, that openmpi-5 switched to using PRRTE.
|
|
# Thus we need to set PRTE_MCA_* instead of OMPI_MCA_*.
|
|
# We keep the openmpi-4 parameters for backward compatability.
|
|
|
|
# Make sure the test starts even if we have less than the requested amount of cores
|
|
export OMPI_MCA_rmaps_base_oversubscribe=1
|
|
export PRTE_MCA_rmaps_default_mapping_policy=node:oversubscribe
|
|
|
|
# Disable CPU pinning
|
|
export OMPI_MCA_hwloc_base_binding_policy=none
|
|
export PRTE_MCA_hwloc_default_binding_policy=none
|
|
;;
|
|
MPICH)
|
|
# Fix to make mpich run in a sandbox
|
|
export HYDRA_IFACE=lo
|
|
;;
|
|
MVAPICH)
|
|
# Disable CPU pinning
|
|
export MV2_ENABLE_AFFINITY=0
|
|
;;
|
|
esac
|
|
|
|
# Limit number of OpenMP threads. Default is "all cores".
|
|
export OMP_NUM_THREADS=1
|
|
}
|
|
|