Merge pull request #171 from JuliaParallel/vc/rand

add rand! for DArrays

Author: vchuravy

src/darray.jl

@@ -784,3 +784,12 @@ function Base.fill!(A::DArray, x)
     end
     return A
 end
+
+using Random
+
+function Random.rand!(rng::AbstractRNG, A::DArray, ::Type{T}) where T
+    asyncmap(procs(A)) do p
+        remotecall_wait((rng, A, T)->rand!(rng, localpart(A), T), p, rng, A, T)
+    end
+end
+

test/darray.jl

@@ -1,6 +1,7 @@
 using Test, LinearAlgebra, SpecialFunctions
 using Statistics: mean
 using SparseArrays: nnz
+using Random
 @everywhere using SparseArrays: sprandn
 
 @testset "test distribute and other constructors" begin
@@ -988,6 +989,13 @@ check_leaks()
     close(d)
 end
 
+@testset "rand!" begin
+    d = dzeros(30, 30)
+    rand!(d)
+
+    close(d)
+end
+
 check_leaks()
 
 d_closeall()