Merge pull request #561 from JuliaGPU/tb/seed

Fix and test seed methods.
JuliaGPU · Jan 8, 2020 · 4dedd0f · 4dedd0f
2 parents e8ff17a + d435e6b
commit 4dedd0f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git 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
@@ -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