Test Open MPI with UCX/OFI

Author: dalcinl

cibw-check-mpi.sh

@@ -117,3 +117,43 @@ if test "$mpiname" = "mpich"; then
     unset MPICH_CH4_NETMOD
     unset MPICH_NOLOCAL
 fi
+
+if test "$mpiname" = "openmpi"; then
+    case $(uname) in
+        Linux)  backends=(ofi ucx) ;;
+        Darwin) backends=(ofi) ;;
+    esac
+    for backend in "${backends[@]}"; do
+        printf "testing %s ... " "$backend"
+        if test "$backend" = "ucx"; then
+            export OMPI_MCA_pml=ucx
+            export OMPI_MCA_osc=ucx
+            export OMPI_MCA_btl=^vader,tcp,openib,uct
+            export OMPI_MCA_opal_common_ucx_tls=tcp
+            export OMPI_MCA_opal_common_ucx_devices=lo
+            export OMPI_MCA_opal_common_ucx_verbose=1
+            check='mca_pml_ucx_init'
+        fi
+        if test "$backend" = "ofi"; then
+            export OMPI_MCA_pml=cm
+            export OMPI_MCA_mtl=ofi
+            export OMPI_MCA_opal_common_ofi_provider_include=tcp
+            export OMPI_MCA_opal_common_ofi_verbose=1
+            check='mtl:ofi:prov.*: tcp'
+        fi
+        mpiexec -n 1 ./helloworld-c 2>&1 | grep -i "$check" > /dev/null
+        for n in $(seq 1 4); do
+            mpiexec -n "$n" ./helloworld-c > /dev/null 2>&1
+        done
+        unset OMPI_MCA_pml
+        unset OMPI_MCA_osc
+        unset OMPI_MCA_mtl
+        unset OMPI_MCA_btl
+        unset OMPI_MCA_opal_common_ucx_tls
+        unset OMPI_MCA_opal_common_ucx_devices
+        unset OMPI_MCA_opal_common_ucx_verbose
+        unset OMPI_MCA_opal_common_ofi_provider_include
+        unset OMPI_MCA_opal_common_ofi_verbose
+        printf "OK\n"
+    done
+fi