Skip to content

Commit 93d2545

Browse files
committed
Support MPIABI
1 parent 1456cae commit 93d2545

2 files changed

Lines changed: 25 additions & 18 deletions

File tree

src/api/api.jl

Lines changed: 18 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -8,18 +8,21 @@ export MPI_Aint, MPI_Count, MPI_Offset, MPI_Status,
88
import MPIPreferences
99
using Libdl
1010

11-
if MPIPreferences.binary == "MPICH_jll"
12-
import MPICH_jll: MPICH_jll, libmpi, libmpi_handle, mpiexec
13-
const libmpiconstants = nothing
14-
elseif MPIPreferences.binary == "OpenMPI_jll"
15-
import OpenMPI_jll: OpenMPI_jll, libmpi, libmpi_handle, mpiexec
11+
if MPIPreferences.binary == "MPIABI_jll"
12+
import MPIABI_jll: MPIABI_jll, libmpi, libmpi_handle, mpiexec
1613
const libmpiconstants = nothing
17-
elseif MPIPreferences.binary == "MicrosoftMPI_jll"
18-
import MicrosoftMPI_jll: MicrosoftMPI_jll, libmpi, libmpi_handle, mpiexec
14+
elseif MPIPreferences.binary == "MPICH_jll"
15+
import MPICH_jll: MPICH_jll, libmpi, libmpi_handle, mpiexec
1916
const libmpiconstants = nothing
2017
elseif MPIPreferences.binary == "MPItrampoline_jll"
2118
import MPItrampoline_jll: MPItrampoline_jll, libmpi, libmpi_handle, mpiexec
2219
const libmpiconstants = MPItrampoline_jll.libload_time_mpi_constants_path
20+
elseif MPIPreferences.binary == "MicrosoftMPI_jll"
21+
import MicrosoftMPI_jll: MicrosoftMPI_jll, libmpi, libmpi_handle, mpiexec
22+
const libmpiconstants = nothing
23+
elseif MPIPreferences.binary == "OpenMPI_jll"
24+
import OpenMPI_jll: OpenMPI_jll, libmpi, libmpi_handle, mpiexec
25+
const libmpiconstants = nothing
2326
elseif MPIPreferences.binary == "system"
2427
import MPIPreferences.System: libmpi, libmpi_handle, mpiexec
2528
const libmpiconstants = nothing
@@ -33,7 +36,7 @@ const initexprs = Any[]
3336
"""
3437
@const_ref name T expr
3538
36-
Defines an constant binding
39+
Defines a constant binding
3740
```julia
3841
const name = Ref{T}()
3942
```
@@ -48,14 +51,16 @@ macro const_ref(name, T, expr)
4851
:(const $(esc(name)) = Ref{$T}())
4952
end
5053

51-
@static if MPIPreferences.abi == "MPICH"
54+
@static if MPIPreferences.abi == "MPIABI"
55+
include("mpiabi.jl")
56+
elseif MPIPreferences.abi == "MPICH"
5257
include("mpich.jl")
53-
elseif MPIPreferences.abi == "OpenMPI"
54-
include("openmpi.jl")
55-
elseif MPIPreferences.abi == "MicrosoftMPI"
56-
include("microsoftmpi.jl")
5758
elseif MPIPreferences.abi == "MPItrampoline"
5859
include("mpitrampoline.jl")
60+
elseif MPIPreferences.abi == "MicrosoftMPI"
61+
include("microsoftmpi.jl")
62+
elseif MPIPreferences.abi == "OpenMPI"
63+
include("openmpi.jl")
5964
elseif MPIPreferences.abi == "HPE MPT"
6065
include("mpt.jl")
6166
else

src/implementations.jl

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -98,14 +98,16 @@ function versioninfo(io::IO=stdout)
9898
println(io, "Package versions")
9999
println(io, " MPI.jl: ", PkgVersion.@Version)
100100
println(io, " MPIPreferences.jl: ", PkgVersion.Version(MPIPreferences))
101-
if MPIPreferences.binary == "MPICH_jll"
101+
if MPIPreferences.binary == "MPIABI_jll"
102+
println(io, " MPIABI_jll: ", PkgVersion.Version(API.MPICH_jll))
103+
elseif MPIPreferences.binary == "MPICH_jll"
102104
println(io, " MPICH_jll: ", PkgVersion.Version(API.MPICH_jll))
103-
elseif MPIPreferences.binary == "OpenMPI_jll"
104-
println(io, " OpenMPI_jll: ", PkgVersion.Version(API.OpenMPI_jll))
105-
elseif MPIPreferences.binary == "MicrosoftMPI_jll"
106-
println(io, " MicrosoftMPI_jll: ", PkgVersion.Version(API.MicrosoftMPI_jll))
107105
elseif MPIPreferences.binary == "MPItrampoline_jll"
108106
println(io, " MPItrampoline_jll ", PkgVersion.Version(API.MPItrampoline_jll))
107+
elseif MPIPreferences.binary == "MicrosoftMPI_jll"
108+
println(io, " MicrosoftMPI_jll: ", PkgVersion.Version(API.MicrosoftMPI_jll))
109+
elseif MPIPreferences.binary == "OpenMPI_jll"
110+
println(io, " OpenMPI_jll: ", PkgVersion.Version(API.OpenMPI_jll))
109111
end
110112

111113
println(io)

0 commit comments

Comments
 (0)