Skip to content

Commit 99d7ead

Browse files
committed
fix
1 parent 3c3f55c commit 99d7ead

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/CUDAKernels.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -197,11 +197,11 @@ end
197197
end
198198

199199
@device_override @inline function KI.get_global_id()
200-
return (; x = Int(blockDim().x), y = Int(blockDim().y), z = Int(blockDim().z))
200+
return (; x = Int((blockIdx().x-1)*blockDim().x + threadIdx().x), y = Int((blockIdx().y-1)*blockDim().y + threadIdx().y), z = Int((blockIdx().z-1)*blockDim().z + threadIdx().z))
201201
end
202202

203203
@device_override @inline function KI.get_local_size()
204-
return (; x = Int((blockDim().x-1)*blockDim().x + threadIdx().x), y = Int((blockDim().y-1)*blockDim().y + threadIdx().y), z = Int((blockDim().z-1)*blockDim().z + threadIdx().z))
204+
return (; x = Int(blockDim().x), y = Int(blockDim().y), z = Int(blockDim().z))
205205
end
206206

207207
@device_override @inline function KI.get_num_groups()

0 commit comments

Comments
 (0)