ensure proper initialization in distribute()

JuliaLang · Feb 7, 2015 · 6861489 · 6861489
1 parent 67636af
commit 6861489
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/base/darray.jl b/base/darray.jl
@@ -166,9 +166,15 @@ function distribute(a::AbstractArray)
     owner = myid()
     rr = RemoteRef()
     put!(rr, a)
-    DArray(size(a)) do I
+    d = DArray(size(a)) do I
         remotecall_fetch(owner, ()->fetch(rr)[I...])
     end
+    # Ensure that all workers have fetched their localparts.
+    # Else a gc in between can recover the RemoteRef rr
+    for chunk in d.chunks
+        wait(chunk)
+    end
+    d
 end
 
 function convert{S,T,N}(::Type{Array{S,N}}, d::DArray{T,N})