rand! for typed arrays – pass the element type to scalar rand call.

JuliaLang · Oct 25, 2014 · 89befaf · 89befaf
1 parent 2cddef3
commit 89befaf
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/base/random.jl b/base/random.jl
@@ -128,15 +128,15 @@ rand(r::AbstractRNG, dims::Dims) = rand!(r, Array(Float64, dims))
 rand(r::AbstractRNG, dims::Int...) = rand(r, dims)
 
 function rand!{T}(A::Array{T})
-    for i=1:length(A)
+    for i = 1:length(A)
         A[i] = rand(T)
     end
     A
 end
 
-function rand!(r::AbstractRNG, A::AbstractArray)
-    for i=1:length(A)
-        @inbounds A[i] = rand(r)
+function rand!{T}(r::AbstractRNG, A::AbstractArray{T})
+    for i = 1:length(A)
+        @inbounds A[i] = rand(r, T)
     end
     A
 end