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