Skip to content

Commit 505d69f

Browse files
authored
Don't use Int32 indices in rand kernels. (#1969)
1 parent de8e4f6 commit 505d69f

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/random.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ function Random.rand!(rng::RNG, A::AnyCuArray)
5151
# grid-stride loop
5252
threadId = threadIdx().x
5353
window = blockDim().x * gridDim().x
54-
offset = (blockIdx().x - 1i32) * blockDim().x
54+
offset = (blockIdx().x - 1) * blockDim().x
5555
while offset < length(A)
5656
i = threadId + offset
5757
if i <= length(A)
@@ -87,8 +87,8 @@ function Random.randn!(rng::RNG, A::AnyCuArray{<:Union{AbstractFloat,Complex{<:A
8787

8888
# grid-stride loop
8989
threadId = threadIdx().x
90-
window = (blockDim().x - 1i32) * gridDim().x
91-
offset = (blockIdx().x - 1i32) * blockDim().x
90+
window = (blockDim().x - 1) * gridDim().x
91+
offset = (blockIdx().x - 1) * blockDim().x
9292
while offset < length(A)
9393
i = threadId + offset
9494
j = threadId + offset + window
@@ -120,8 +120,8 @@ function Random.randn!(rng::RNG, A::AnyCuArray{<:Union{AbstractFloat,Complex{<:A
120120

121121
# grid-stride loop
122122
threadId = threadIdx().x
123-
window = (blockDim().x - 1i32) * gridDim().x
124-
offset = (blockIdx().x - 1i32) * blockDim().x
123+
window = (blockDim().x - 1) * gridDim().x
124+
offset = (blockIdx().x - 1) * blockDim().x
125125
while offset < length(A)
126126
i = threadId + offset
127127
if i <= length(A)

0 commit comments

Comments
 (0)