Skip to content

Commit 51b07cd

Browse files
committed
Add in-place rand for CPU arrays.
1 parent 77fbc3b commit 51b07cd

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/random.jl

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,12 @@ function Random.rand!(rng::RNG, A::AnyCuArray)
7878
A
7979
end
8080

81+
function Random.rand!(rng::RNG, A::AbstractArray{T}) where {T}
82+
B = CuArray{T}(undef, size(A))
83+
Random.rand!(rng, B)
84+
copyto!(A, B)
85+
end
86+
8187
function Random.rand(rng::RNG, T::Type)
8288
assertscalar("scalar rand")
8389
A = CuArray{T}(undef, 1)

0 commit comments

Comments
 (0)