From d435e6b1567a11f5683cb66e125978eb40b378ac Mon Sep 17 00:00:00 2001 From: Tim Besard Date: Wed, 8 Jan 2020 11:23:06 +0100 Subject: [PATCH] Fix and test seed methods. --- src/rand/random.jl | 2 +- test/rand.jl | 7 +++++++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/src/rand/random.jl b/src/rand/random.jl index bd90564f..79b04b6f 100644 --- a/src/rand/random.jl +++ b/src/rand/random.jl @@ -42,7 +42,7 @@ Base.unsafe_convert(::Type{curandGenerator_t}, rng::RNG) = rng.handle seed!(rng::RNG=generator()) = (curandGenerateSeeds(rng); return) seed!(seed::Int64, offset::Int64=0) = seed!(generator(), seed, offset) -function seed!(rng::RNG, seed::Int64, offset::Int64) +function seed!(rng::RNG, seed::Int64, offset::Int64=0) curandSetPseudoRandomGeneratorSeed(rng, seed) curandSetGeneratorOffset(rng, offset) @allocates curandGenerateSeeds(rng) diff --git a/test/rand.jl b/test/rand.jl index 6cb93305..8972fa73 100644 --- a/test/rand.jl +++ b/test/rand.jl @@ -3,6 +3,13 @@ using CuArrays.CURAND CURAND.seed!() +CURAND.seed!(1) +CURAND.seed!(1, 0) + +rng = CURAND.generator() +CURAND.seed!(rng) +CURAND.seed!(rng, 1) +CURAND.seed!(rng, 1, 0) # NOTE: tests should cover both pow2 and non-pow2 dims