Merge pull request #103 from JuliaParallel/anj/copy

Allow different chunk sizes in the two arrays in copy!

Author: vchuravy

src/core.jl

@@ -551,14 +551,10 @@ Base.getindex(d::DArray) = d[1]
 Base.getindex(d::DArray, I::Union{Int,UnitRange{Int},Colon,Vector{Int},StepRange{Int,Int}}...) = view(d, I...)
 
 Base.copy!(dest::SubOrDArray, src::SubOrDArray) = begin
-    if !(size(dest) == size(src) &&
-         procs(dest) == procs(src) &&
-         dest.indexes == src.indexes &&
-         dest.cuts == src.cuts)
-        throw(DimensionMismatch("destination array doesn't fit to source array"))
-    end
-    @sync for p in procs(dest)
-        @async remotecall_fetch((dest,src)->(copy!(localpart(dest), localpart(src)); nothing), p, dest, src)
+    asyncmap(procs(dest)) do p
+        remotecall_fetch(p) do
+            localpart(dest)[:] = src[localindexes(dest)...]
+        end
     end
     return dest
 end

test/darray.jl

@@ -154,6 +154,19 @@ end
 
 check_leaks()
 
+@testset "copy!" begin
+    D1 = dzeros((10,10))
+    r1 = remotecall_wait(() -> randn(3,10), workers()[1])
+    r2 = remotecall_wait(() -> randn(7,10), workers()[2])
+    D2 = DArray(reshape([r1; r2], 2, 1))
+    copy!(D2, D1)
+    @test D1 == D2
+    close(D1)
+    close(D2)
+end
+
+check_leaks()
+
 @testset "test DArray reduce" begin
     D = DArray(id->fill(myid(), map(length,id)), (10,10), [MYID, OTHERIDS])