Native RNG: Fix randn window calculation.

JuliaGPU · Nov 21, 2024 · 2775778 · 2775778
1 parent b7bee72
commit 2775778
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/random.jl b/src/random.jl
@@ -96,7 +96,7 @@ function Random.randn!(rng::RNG, A::AnyCuArray{<:Union{AbstractFloat,Complex{<:A
 
         # grid-stride loop
         threadId = threadIdx().x
-        window = widemul(blockDim().x - 1i32, gridDim().x)
+        window = widemul(blockDim().x, gridDim().x)
         offset = widemul(blockIdx().x - 1i32, blockDim().x)
         while offset < length(A)
             i = threadId + offset
@@ -129,7 +129,7 @@ function Random.randn!(rng::RNG, A::AnyCuArray{<:Union{AbstractFloat,Complex{<:A
 
         # grid-stride loop
         threadId = threadIdx().x
-        window = widemul(blockDim().x - 1i32, gridDim().x)
+        window = widemul(blockDim().x, gridDim().x)
         offset = widemul(blockIdx().x - 1i32, blockDim().x)
         while offset < length(A)
             i = threadId + offset