Skip to content

Commit 0bbeb97

Browse files
committed
Native RNG: Fix randn window calculation.
1 parent 0a515fc commit 0bbeb97

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/random.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ function Random.randn!(rng::RNG, A::AnyCuArray{<:Union{AbstractFloat,Complex{<:A
9696

9797
# grid-stride loop
9898
threadId = threadIdx().x
99-
window = widemul(blockDim().x - 1i32, gridDim().x)
99+
window = widemul(blockDim().x, gridDim().x)
100100
offset = widemul(blockIdx().x - 1i32, blockDim().x)
101101
while offset < length(A)
102102
i = threadId + offset
@@ -129,7 +129,7 @@ function Random.randn!(rng::RNG, A::AnyCuArray{<:Union{AbstractFloat,Complex{<:A
129129

130130
# grid-stride loop
131131
threadId = threadIdx().x
132-
window = widemul(blockDim().x - 1i32, gridDim().x)
132+
window = widemul(blockDim().x, gridDim().x)
133133
offset = widemul(blockIdx().x - 1i32, blockDim().x)
134134
while offset < length(A)
135135
i = threadId + offset

0 commit comments

Comments
 (0)