action: Support alternative MPI names

openmpi: open-mpi, ompi intelmpi: intel-mpi, impi, intel msmpi: microsoft-mpi, microsoftmpi, microsoft
mpi4py · Oct 9, 2023 · a2cf984 · a2cf984
1 parent 80d7f14
commit a2cf984
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/setup-mpi.sh b/setup-mpi.sh
@@ -1,7 +1,22 @@
 #!/bin/bash
 set -eu
 
-MPI=$(echo "${1:-}" | tr '[:upper:]' '[:lower:]')
+MPI=$(echo "${1:-}" | tr '[:upper:]' '[:lower:]' | tr -d '-')
+
+case $MPI in
+    mpich)
+        MPI=mpich
+        ;;
+    ompi | openmpi)
+        MPI=openmpi
+        ;;
+    impi | intelmpi | intel)
+        MPI=intelmpi
+        ;;
+    msmpi | microsoftmpi | microsoft)
+        MPI=msmpi
+        ;;
+esac
 
 setup-apt-intel-oneapi () {
     apt_repo_url=https://apt.repos.intel.com/