Add MPICHABI support for Cray MPICH (#379)

* Add MPICHABI support for Cray MPICH Co-authored-by: Samuel Omlin <[email protected]> Co-authored-by: Simon Byrne <[email protected]>
JuliaParallel · May 8, 2020 · 1971a5c · 1971a5c
1 parent 3e52a3a
commit 1971a5c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/deps/build.jl b/deps/build.jl
@@ -86,7 +86,9 @@ if binary == "system"
         impl, version = identify_implementation()
         if (impl == MPICH && version >= v"3.1" ||
             impl == IntelMPI && version > v"2014" ||
-            impl == MVAPICH && version >= v"2")
+            impl == MVAPICH && version >= v"2" ||
+            impl == CrayMPICH && version >= v"7")
+            # https://www.mpich.org/abi/
             abi = "MPICH"
         elseif impl == OpenMPI || impl == IBMSpectrumMPI
             abi = "OpenMPI"

diff --git a/src/implementations.jl b/src/implementations.jl
@@ -48,6 +48,7 @@ An enum corresponding to known MPI implementations
 - `IntelMPI`: [Intel MPI](https://software.intel.com/en-us/mpi-library)
 - `SpectrimMPI`: [IBM Spectrum MPI](https://www.ibm.com/us-en/marketplace/spectrum-mpi)
 - `MVAPICH`: [MVAPICH](http://mvapich.cse.ohio-state.edu/)
+- `CrayMPICH`: Part of the Cray Message Passing Toolkit (MPT)
 
 # See also
 
@@ -61,6 +62,7 @@ An enum corresponding to known MPI implementations
     IntelMPI
     IBMSpectrumMPI
     MVAPICH
+    CrayMPICH
 end
 
 """
@@ -117,6 +119,13 @@ function identify_implementation()
         if (m = match(r"^MVAPICH2? Version\s*:\t(\S*)\n", MPI_LIBRARY_VERSION_STRING)) !== nothing
             version = VersionNumber(m.captures[1])
         end
+
+    elseif occursin("CRAY MPICH", MPI_LIBRARY_VERSION_STRING)
+        impl = CrayMPICH
+        # "MPI VERSION    : CRAY MPICH version 7.7.10 (ANL base 3.2)\n"
+        if (m = match(r"CRAY MPICH version (\d+.\d+.\d+)", MPI_LIBRARY_VERSION_STRING)) !== nothing
+            version = VersionNumber(m.captures[1])
+        end
     end
 
     return impl, version