Merge branch 'master' into patch-1

Author: dkarrasch

Manifest.toml

@@ -81,9 +81,9 @@ version = "0.1.4"
 
 [[GPUCompiler]]
 deps = ["ExprTools", "InteractiveUtils", "LLVM", "Libdl", "Logging", "Scratch", "TimerOutputs", "UUIDs"]
-git-tree-sha1 = "550bb5127b9b6cf04bb86d72ac37a81a11a204d6"
+git-tree-sha1 = "11b2d77f29a85f3649c273a38f6618121c6b1c51"
 uuid = "61eb1bfa-7361-4325-ad38-22787b887f55"
-version = "0.20.1"
+version = "0.20.2"
 
 [[InteractiveUtils]]
 deps = ["Markdown"]

src/random.jl

@@ -1,4 +1,22 @@
 using Random
 
-Random.rand!(A::oneWrappedArray) = Random.rand!(GPUArrays.default_rng(oneArray), A)
-Random.randn!(A::oneWrappedArray) = Random.randn!(GPUArrays.default_rng(oneArray), A)
+gpuarrays_rng() = GPUArrays.default_rng(oneArray)
+
+# GPUArrays in-place
+Random.rand!(A::oneWrappedArray) = Random.rand!(gpuarrays_rng(), A)
+Random.randn!(A::oneWrappedArray) = Random.randn!(gpuarrays_rng(), A)
+
+# GPUArrays out-of-place
+rand(T::Type, dims::Dims) = Random.rand!(oneArray{T}(undef, dims...))
+randn(T::Type, dims::Dims; kwargs...) = Random.randn!(oneArray{T}(undef, dims...); kwargs...)
+
+# support all dimension specifications
+rand(T::Type, dim1::Integer, dims::Integer...) = Random.rand!(oneArray{T}(undef, dim1, dims...))
+randn(T::Type, dim1::Integer, dims::Integer...; kwargs...) = Random.randn!(oneArray{T}(undef, dim1, dims...); kwargs...)
+
+# untyped out-of-place
+rand(dim1::Integer, dims::Integer...) = Random.rand!(oneArray{Float32}(undef, dim1, dims...))
+randn(dim1::Integer, dims::Integer...; kwargs...) = Random.randn!(oneArray{Float32}(undef, dim1, dims...); kwargs...)
+
+# seeding
+seed!(seed=Base.rand(UInt64)) = Random.seed!(gpuarrays_rng(), seed)

test/random.jl

@@ -0,0 +1,39 @@
+using Random
+
+@testset "rand" begin
+
+# in-place
+for (f,T) in ((rand!,Float16),
+              (rand!,Float32),
+              (randn!,Float16),
+              (randn!,Float32)),
+    d in (2, (2,2), (2,2,2), 3, (3,3), (3,3,3))
+    A = oneArray{T}(undef, d)
+    fill!(A, T(0))
+    f(A)
+    @test !iszero(collect(A))
+end
+
+# out-of-place, with implicit type
+for (f,T) in ((oneAPI.rand,Float32), (oneAPI.randn,Float32)),
+    args in ((2,), (2, 2), (3,), (3, 3))
+    A = f(args...)
+    @test eltype(A) == T
+end
+
+# out-of-place, with type specified
+for (f,T) in ((oneAPI.rand,Float32), (oneAPI.randn,Float32),
+              (rand,Float32), (randn,Float32)),
+    args in ((T, 2), (T, 2, 2), (T, (2, 2)), (T, 3), (T, 3, 3), (T, (3, 3)))
+    A = f(args...)
+    @test eltype(A) == T
+end
+
+## seeding
+oneAPI.seed!(1)
+a = oneAPI.rand(Int32, 1)
+oneAPI.seed!(1)
+b = oneAPI.rand(Int32, 1)
+@test iszero(collect(a) - collect(b))
+
+end # testset