Add wrappers for MPI_COMM_SPLIT and MPI_COMM_SPLIT_TYPE for defining new communicators, especially node-wise communicators and constants.

Add preliminary wrappers for MPI_Info type, which is required by MPI_COMM_SPLIT_TYPE.
Add wrappers for Wtime and Wtick.

Author: tbole

deps/gen_constants.f90

@@ -28,6 +28,8 @@ program gen_constants
   call output("MPI_COMM_SELF   ", MPI_COMM_SELF)
   call output("MPI_COMM_WORLD  ", MPI_COMM_WORLD)
 
+  call output("MPI_COMM_TYPE_SHARED", MPI_COMM_TYPE_SHARED)
+
   call output("MPI_OP_NULL     ", MPI_OP_NULL)
   call output("MPI_BAND        ", MPI_BAND)
   call output("MPI_BOR         ", MPI_BOR)
@@ -45,6 +47,8 @@ program gen_constants
 
   call output("MPI_REQUEST_NULL", MPI_REQUEST_NULL)
 
+  call output("MPI_INFO_NULL   ", MPI_INFO_NULL)
+
   call output("MPI_STATUS_SIZE ", MPI_STATUS_SIZE)
   call output("MPI_ERROR       ", MPI_ERROR)
   call output("MPI_SOURCE      ", MPI_SOURCE)

deps/gen_functions.c

@@ -32,6 +32,7 @@ int main(int argc, char *argv[]) {
   printf("    :MPI_COMM_RANK          => \"%s\",\n", STRING(MPI_COMM_RANK));
   printf("    :MPI_COMM_SIZE          => \"%s\",\n", STRING(MPI_COMM_SIZE));
   printf("    :MPI_COMM_SPLIT         => \"%s\",\n", STRING(MPI_COMM_SPLIT));
+  printf("    :MPI_COMM_SPLIT_TYPE    => \"%s\",\n", STRING(MPI_COMM_SPLIT_TYPE));
   printf("    :MPI_EXSCAN             => \"%s\",\n", STRING(MPI_EXSCAN));
   printf("    :MPI_FINALIZE           => \"%s\",\n", STRING(MPI_FINALIZE));
   printf("    :MPI_FINALIZED          => \"%s\",\n", STRING(MPI_FINALIZED));
@@ -70,6 +71,7 @@ int main(int argc, char *argv[]) {
   printf("    :MPI_WAITALL            => \"%s\",\n", STRING(MPI_WAITALL));
   printf("    :MPI_WAITANY            => \"%s\",\n", STRING(MPI_WAITANY));
   printf("    :MPI_WAITSOME           => \"%s\",\n", STRING(MPI_WAITSOME));
+  printf("    :MPI_WTICK              => \"%s\",\n", STRING(MPI_WTICK));
   printf("    :MPI_WTIME              => \"%s\",\n", STRING(MPI_WTIME));
   printf("    :MPI_TYPE_CREATE_STRUCT => \"%s\",\n", 
          STRING(MPI_TYPE_CREATE_STRUCT));

src/mpi-base.jl

@@ -93,6 +93,11 @@ mutable struct Request
 end
 const REQUEST_NULL = Request(MPI_REQUEST_NULL, nothing)
 
+mutable struct Info
+    val::Cint
+end
+const INFO_NULL = Info(MPI_INFO_NULL)
+
 mutable struct Status
     val::Array{Cint,1}
     Status() = new(Array{Cint}(MPI_STATUS_SIZE))
@@ -169,6 +174,28 @@ function Comm_size(comm::Comm)
     Int(size[])
 end
 
+function Comm_split(comm::Comm,color::Integer,key::Integer)
+    newcomm = Ref{Cint}()
+    ccall(MPI_COMM_SPLIT, Void, (Ptr{Cint}, Ptr{Cint}, Ptr{Cint}, Ptr{Cint}, Ptr{Cint}),
+          &comm.val, &color, &key, newcomm, &0)
+    MPI.Comm(newcomm[])
+end
+
+function Comm_split_type(comm::Comm,split_type::Integer,key::Integer;info::Info=INFO_NULL)
+    newcomm = Ref{Cint}()
+    ccall(MPI_COMM_SPLIT_TYPE, Void, (Ptr{Cint}, Ptr{Cint}, Ptr{Cint}, Ptr{Cint}, Ptr{Cint}, Ptr{Cint}),
+          &comm.val, &split_type, &key, &info.val, newcomm, &0)
+    MPI.Comm(newcomm[])
+end
+
+function Wtick()
+    ccall(MPI_WTICK, Cdouble, () )
+end
+
+function Wtime()
+    ccall(MPI_WTIME, Cdouble, () )
+end
+
 function type_create(T::DataType)
     if !isbits(T)
         throw(ArgumentError("Type must be isbits()"))

src/win_mpiconstants.jl

@@ -22,6 +22,7 @@ const MPI_INTEGER8 = Int32(0x4c000831)
 const MPI_COMM_NULL = Int32(67108864)
 const MPI_COMM_SELF = Int32(1140850689)
 const MPI_COMM_WORLD = Int32(1140850688)
+const MPI_COMM_TYPE_SHARED = Int32(1)
 const MPI_OP_NULL = Int32(402653184)
 const MPI_BAND = Int32(1476395014)
 const MPI_BOR = Int32(1476395016)
@@ -37,6 +38,7 @@ const MPI_PROD = Int32(1476395012)
 const MPI_REPLACE = Int32(1476395021)
 const MPI_SUM = Int32(1476395011)
 const MPI_REQUEST_NULL = Int32(738197504)
+const MPI_INFO_NULL = Int32(469762048)
 const MPI_STATUS_SIZE = Int32(5)
 const MPI_ERROR = Int32(5)
 const MPI_SOURCE = Int32(3)
@@ -51,6 +53,8 @@ const MPI_ABORT = (:MPI_ABORT, "msmpi.dll")
 const MPI_INIT = (:MPI_INIT, "msmpi.dll")
 const MPI_COMM_RANK = (:MPI_COMM_RANK, "msmpi.dll")
 const MPI_COMM_SIZE = (:MPI_COMM_SIZE, "msmpi.dll")
+const MPI_COMM_SPLIT = (:MPI_COMM_SPLIT, "msmpi.dll")
+const MPI_COMM_SPLIT_TYPE = (:MPI_COMM_SPLIT_TYPE, "msmpi.dll")
 const MPI_BARRIER = (:MPI_BARRIER, "msmpi.dll")
 const MPI_FINALIZE = (:MPI_FINALIZE, "msmpi.dll")
 const MPI_BCAST = (:MPI_BCAST, "msmpi.dll")