Open MPI: Tweak MCA params

mpi4py · Apr 8, 2021 · c58ae27 · c58ae27
1 parent 7647f39
commit c58ae27
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/setup-mpi.sh b/setup-mpi.sh
@@ -61,3 +61,18 @@ case $(uname) in
 esac
 
 echo "::set-output name=mpi::${MPI}"
+
+if [ $MPI == openmpi ]; then
+    openmpi_mca_params=$HOME/.openmpi/mca-params.conf
+    mkdir -p $(dirname $openmpi_mca_params)
+    echo plm=isolated >> $openmpi_mca_params
+    echo rmaps_base_oversubscribe=true >> $openmpi_mca_params
+    echo btl_base_warn_component_unused=false >> $openmpi_mca_params
+    echo btl_vader_single_copy_mechanism=none >> $openmpi_mca_params
+    if [[ $(uname) == Darwin ]]; then
+        # open-mpi/ompi#7516
+        echo gds=hash >> $openmpi_mca_params
+        # open-mpi/ompi#5798
+        echo btl_vader_backing_directory=/tmp >> $openmpi_mca_params
+    fi
+fi