Skip to content
This repository was archived by the owner on Mar 12, 2021. It is now read-only.

Commit d435e6b

Browse files
committed
Fix and test seed methods.
1 parent e8ff17a commit d435e6b

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

src/rand/random.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ Base.unsafe_convert(::Type{curandGenerator_t}, rng::RNG) = rng.handle
4242
seed!(rng::RNG=generator()) = (curandGenerateSeeds(rng); return)
4343

4444
seed!(seed::Int64, offset::Int64=0) = seed!(generator(), seed, offset)
45-
function seed!(rng::RNG, seed::Int64, offset::Int64)
45+
function seed!(rng::RNG, seed::Int64, offset::Int64=0)
4646
curandSetPseudoRandomGeneratorSeed(rng, seed)
4747
curandSetGeneratorOffset(rng, offset)
4848
@allocates curandGenerateSeeds(rng)

test/rand.jl

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,13 @@
33
using CuArrays.CURAND
44

55
CURAND.seed!()
6+
CURAND.seed!(1)
7+
CURAND.seed!(1, 0)
8+
9+
rng = CURAND.generator()
10+
CURAND.seed!(rng)
11+
CURAND.seed!(rng, 1)
12+
CURAND.seed!(rng, 1, 0)
613

714
# NOTE: tests should cover both pow2 and non-pow2 dims
815

0 commit comments

Comments
 (0)