Support for Open MPI

Author: dalcinl

.codespellrc

@@ -0,0 +1,4 @@
+# https://github.com/codespell-project/codespell #-*- mode: conf -*-
+
+[codespell]
+ignore-words-list = pevent

.github/workflows/cd-wheel.yml

@@ -7,12 +7,12 @@ on:  # yamllint disable-line rule:truthy
   workflow_call:
     inputs:
       mpiname:
-        description: 'MPI'
+        description: 'MPI Name'
         default: 'mpich'
         required: false
         type: string
       version:
-        description: 'Version'
+        description: 'MPI Version'
         default: ''
         required: false
         type: string
@@ -39,6 +39,7 @@ on:  # yamllint disable-line rule:truthy
         type: choice
         options:
           - mpich
+          - openmpi
       version:
         description: 'Version'
         default: ''
@@ -122,6 +123,11 @@ jobs:
     - id: checkout
       uses: actions/checkout@v4
 
+    - id: setup-autotools
+      if: ${{ runner.os == 'macOS' && runner.arch == 'ARM64' }}
+      run: |
+        brew install autoconf automake libtool
+
     - id: setup-gfortran
       if: ${{ runner.os == 'macOS' }}
       run: |

.github/workflows/cd.yml

@@ -7,13 +7,14 @@ on:  # yamllint disable-line rule:truthy
   workflow_dispatch:
     inputs:
       mpiname:
-        description: 'MPI'
+        description: 'MPI Name'
         default: 'mpich'
         type: choice
         options:
           - mpich
+          - openmpi
       version:
-        description: 'Version'
+        description: 'MPI Version'
         default: ''
         required: false
         type: string

README.md

@@ -1,29 +1,33 @@
-# mpich-publish
+# mpi-publish
 
-This repository builds and publishes [MPICH](https://mpich.org)
-Python wheels able to run in a variety of
+This repository builds and publishes [MPICH](https://www.mpich.org) and
+[Open MPI](https://www.open-mpi.org/) Python wheels able to run in
+a variety of
 
 - operating systems: *Linux*, *macOS*;
 - processor architectures: *AMD64*, *ARM64*, *PPC64*;
 - Python implementations: *CPython*, *PyPy*.
 
-MPICH wheels are uploaded to the [Anaconda.org](https://anaconda.org/mpi4py)
+MPI wheels are uploaded to the [Anaconda.org](https://anaconda.org/mpi4py)
 package server. These wheels can be installed with `pip` specifying the
 alternative index URL:
 
 ```sh
 python -m pip install mpich -i https://pypi.anaconda.org/mpi4py/simple
 ```
 
+```sh
+python -m pip install openmpi -i https://pypi.anaconda.org/mpi4py/simple
+```
+
 > [!CAUTION]
-> MPICH wheels are distributed with a focus on ease of use, compatibility,
-> and interoperability. The Linux MPICH wheels are built in somewhat
-> constrained environments with relatively dated Linux distributions
+> MPI wheels are distributed with a focus on ease of use, compatibility, and
+> interoperability. The Linux wheels are built in somewhat constrained
+> environments with relatively dated Linux distributions
 > ([manylinux](https://github.com/pypa/manylinux) container images).
 > Therefore, they may lack support for high-performance features like
 > cross-memory attach (XPMEM/CMA). In production scenarios, it is recommended
-> to use external (either custom-built or system-provided) MPICH
-> installations.
+> to use external (either custom-built or system-provided) MPI installations.
 
 > [!TIP]
 > [Intel MPI](https://software.intel.com/intel-mpi-library) distributes [Linux

bootstrap.sh

@@ -3,14 +3,24 @@ set -euo pipefail
 
 mpiname=${MPINAME:-mpich}
 
+PROJECT=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
+PACKAGE=$PROJECT/package
+SOURCE=$PACKAGE/source
+
 if test "$mpiname" = "mpich"; then
     version=${VERSION:-4.2.0}
     urlbase="https://www.mpich.org/static/downloads/$version"
     tarball="$mpiname-$version.tar.gz"
     license=COPYRIGHT
 fi
 
-SOURCE=package/source
+if test "$mpiname" = "openmpi"; then
+    version=${VERSION:-5.0.2}
+    urlbase=https://download.open-mpi.org/release/open-mpi/v${version%.*}
+    tarball="$mpiname-$version.tar.gz"
+    license=LICENSE
+fi
+
 if test ! -d "$SOURCE"; then
     if test ! -f "$tarball"; then
         echo downloading "$urlbase"/"$tarball"...
@@ -21,6 +31,15 @@ if test ! -d "$SOURCE"; then
     echo extracting "$tarball" to "$SOURCE"...
     tar xf "$tarball"
     mv "$mpiname-$version" "$SOURCE"
+    if test "$mpiname-$(uname)" = "openmpi-Darwin"; then
+        if test -d "$SOURCE"/3rd-party; then
+            cd "$SOURCE"/3rd-party
+            tar xf libevent-*.tar.gz && cd libevent-*
+            echo running autogen.sh on "$(basename "$(pwd)")"
+            ./autogen.sh
+            cd "$PROJECT"
+        fi
+    fi
 else
     echo reusing directory "$SOURCE"...
 fi

check-wheel.sh

@@ -56,6 +56,7 @@ if test "$mpiname" = "mpich"; then
         "$data"/bin/mpic++
         "$data"/bin/mpicxx
     )
+    wrappers=()
     programs=(
         "$data"/bin/mpichversion
         "$data"/bin/mpivars
@@ -74,7 +75,32 @@ if test "$mpiname" = "mpich"; then
     fi
 fi
 
-check-dso() {
+if test "$mpiname" = "openmpi"; then
+    headers=(
+        "$data"/include/mpi.h
+    )
+    scripts=()
+    wrappers=(
+        "$data"/bin/mpicc
+        "$data"/bin/mpic++
+        "$data"/bin/mpicxx
+        "$data"/bin/mpiCC
+        "$data"/bin/mpirun
+        "$data"/bin/mpiexec
+    )
+    programs=(
+        "$data"/bin/*_info
+        "$data"/bin/*_wrapper
+    )
+    libraries=(
+        "$data"/lib/libmpi.*
+        "$data"/lib/libopen-*.*
+    )
+    runlibs+='|lib(z|util|event.*|hwloc)'$soregex
+    runlibs+='|lib(open-(pal|rte)|pmix|prrte)'$soregex
+fi
+
+check-binary() {
     local dso=$1 out1="" out2=""
     echo checking "$dso"...
     test -f "$dso" || printf "ERROR: file not found\n"
@@ -85,25 +111,33 @@ check-dso() {
     test -z "$out1" -a -z "$out2"
 }
 
-for header in "${headers[@]}"; do
+for header in "${headers[@]-}"; do
+    test -n "$header" || break
     echo checking "$header"...
     test -f "$header"
     out=$(grep -E '^#include\s+"mpicxx\.h"' "$header" || true)
     test -z "$out" || printf "ERROR: include\n%s\n" "$out"
     test -z "$out"
 done
-for script in "${scripts[@]}"; do
+for script in "${scripts[@]-}"; do
+    test -n "$script" || break
     echo checking "$script"...
     test -f "$script"
     out=$(grep -E "/opt/$mpiname" "$script" || true)
     test -z "$out" || printf "ERROR: prefix\n%s\n" "$out"
     test -z "$out"
 done
-for bin in "${programs[@]}"; do
-    check-dso "$bin"
+for bin in "${wrappers[@]-}"; do
+    test -n "$bin" || break
+    check-binary "$bin"
+done
+for bin in "${programs[@]-}"; do
+    test -n "$bin" || break
+    check-binary "$bin"
 done
-for lib in "${libraries[@]}"; do
-    check-dso "$lib"
+for lib in "${libraries[@]-}"; do
+    test -n "$lib" || break
+    check-binary "$lib"
 done
 if test "$(uname)" = Linux; then
     echo checking "$pkgname".libs...