Wrapped Ibcast

Author: kshyatt

deps/CMakeLists.txt

@@ -50,6 +50,7 @@ FortranCInterface_HEADER(jlmpi_f2c.h MACRO_NAMESPACE "JLMPI_" SYMBOLS
   MPI_GET_COUNT
   MPI_GET_PROCESSOR_NAME
   MPI_IBARRIER
+  MPI_IBCAST
   MPI_INIT
   MPI_INITIALIZED
   MPI_IPROBE

deps/gen_functions.c

@@ -38,6 +38,7 @@ int main(int argc, char *argv[]) {
   printf("    :MPI_GET_PROCESSOR_NAME => \"%s\",\n",
          STRING(MPI_GET_PROCESSOR_NAME));
   printf("    :MPI_IBARRIER           => \"%s\",\n", STRING(MPI_IBARRIER));
+  printf("    :MPI_IBCAST             => \"%s\",\n", STRING(MPI_IBCAST));
   printf("    :MPI_INIT               => \"%s\",\n", STRING(MPI_INIT));
   printf("    :MPI_INITIALIZED        => \"%s\",\n", STRING(MPI_INITIALIZED));
   printf("    :MPI_IPROBE             => \"%s\",\n", STRING(MPI_IPROBE));

src/mpi-base.jl

@@ -485,6 +485,19 @@ function Bcast!{T}(buffer::Array{T}, root::Integer, comm::Comm)
     Bcast!(buffer, length(buffer), root, comm)
 end
 
+function Ibcast!{T}(buffer::MPIBuffertype{T}, count::Integer,
+                   root::Integer, comm::Comm)
+    rval = Ref{Cint}()
+    ccall(MPI_IBCAST, Void,
+          (Ptr{T}, Ptr{Cint}, Ptr{Cint}, Ptr{Cint}, Ptr{Cint}, Ptr{Cint}, Ptr{Cint}),
+          buffer, &count, &mpitype(T), &root, &comm.val, rval, &0)
+    Request(rval[], buffer), buffer
+end
+
+function Ibcast!{T}(buffer::Array{T}, root::Integer, comm::Comm)
+    Ibcast!(buffer, length(buffer), root, comm)
+end
+
 #=
 function Bcast{T}(obj::T, root::Integer, comm::Comm)
     buf = [T]

src/win_mpiconstants.jl

@@ -54,6 +54,7 @@ const MPI_BARRIER = (:MPI_BARRIER, "msmpi.dll")
 const MPI_IBARRIER = (:MPI_IBARRIER, "msmpi.dll")
 const MPI_FINALIZE = (:MPI_FINALIZE, "msmpi.dll")
 const MPI_BCAST = (:MPI_BCAST, "msmpi.dll")
+const MPI_IBCAST = (:MPI_IBCAST, "msmpi.dll")
 const MPI_REDUCE = (:MPI_REDUCE, "msmpi.dll")
 const MPI_IRECV = (:MPI_IRECV, "msmpi.dll")
 const MPI_RECV = (:MPI_RECV, "msmpi.dll")

test/test_bcast.jl

@@ -16,6 +16,19 @@ function bcast_array(A, root)
     B
 end
 
+function ibcast_array(A, root)
+    comm = MPI.COMM_WORLD
+
+    if MPI.Comm_rank(comm) == root
+        B = copy(A)
+    else
+        B = similar(A)
+    end
+
+    req, B = MPI.Ibcast!(B, root, comm)
+    req, B
+end
+
 root = 0
 
 srand(17)
@@ -53,4 +66,11 @@ end
 B = MPI.bcast(B, root, comm)
 @test B["foo"] == "bar"
 
+#ibcast
+A = ['s', 't', 'a', 'r', ' ', 'w', 'a', 'r', 's']
+req, B = ibcast_array(A, root)
+MPI.Wait!(req)
+MPI.Barrier(MPI.COMM_WORLD)
+@test B == A
+
 MPI.Finalize()