Use MPICH instead of OpenMPI

Possibly a packaging bug in Ubuntu 26.04 with OpenMPI:

$ mpirun --version
--------------------------------------------------------------------------
Sorry!  You were supposed to get help about:
    version
But I couldn't open the help file:
    /usr/lib/x86_64-linux-gnu/pmix2/share/pmix/help-schizo-ompi.txt: No such file or directory
    /usr/lib/x86_64-linux-gnu/prrte3/share/prte/help-schizo-ompi.txt: No such file or directory.
Sorry!
--------------------------------------------------------------------------

Author: amandalund

scripts/collect_metadata.py

@@ -54,13 +54,17 @@ def get_env_vars() -> dict:
 
 
 def get_mpi_version() -> str:
-    """Get MPI version from ompi_info."""
+    """Get MPI version string from mpirun."""
     try:
-        return subprocess.check_output(
-            ["ompi_info", "--version"],
+        output = subprocess.check_output(
+            ["mpirun", "--version"],
             stderr=subprocess.STDOUT,
             text=True,
-        ).splitlines()[0].strip()
+        )
+        for line in output.splitlines():
+            if "Version:" in line:
+                return line.split(":", 1)[1].strip()
+        return output.splitlines()[0].strip()
     except Exception:
         return "unknown"
 

scripts/setup_ami.sh

@@ -14,7 +14,7 @@ sudo DEBIAN_FRONTEND=noninteractive apt-get upgrade -y
 sudo DEBIAN_FRONTEND=noninteractive apt-get install -y \
   build-essential g++ gfortran cmake git wget jq \
   python${PYVER} python${PYVER}-venv python${PYVER}-dev python3-pip \
-  openmpi-bin libopenmpi-dev libhdf5-openmpi-dev libpng-dev libpugixml-dev \
+  mpich libmpich-dev libhdf5-mpich-dev libpng-dev libpugixml-dev \
   libfmt-dev libeigen3-dev libnetcdf-dev libtbb-dev numactl libembree-dev time
 
 sudo mkdir -p ${TOOLS_VENV}