Skip to content

Commit 865b6d8

Browse files
committed
Wait for GPU events using synchronize
1 parent 4911918 commit 865b6d8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/backends/cuda.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ struct CudaEvent <: Event
4949
end
5050
function wait(ev::CudaEvent, progress=nothing)
5151
if progress === nothing
52-
CUDAdrv.wait(ev.event)
52+
CUDAdrv.synchronize(ev.event)
5353
else
5454
while !CUDAdrv.query(ev.event)
5555
progress()
@@ -239,4 +239,4 @@ end
239239
ptr = pointer(arr) + (I[1].start-1)*sizeof(T)
240240
len = I[1].stop - I[1].start + 1
241241
return ConstCuDeviceArray{T,1,A}(len, ptr)
242-
end
242+
end

0 commit comments

Comments
 (0)