Skip to content

Commit 7770a63

Browse files
committed
fix viewable return & event leak
1 parent 60682de commit 7770a63

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

cuda_py/cuda/py/_memoryview.pyx

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -263,6 +263,7 @@ cdef GPUMemoryView view_as_cai(obj, stream_ptr):
263263
cuda.CUevent_flags.CU_EVENT_DISABLE_TIMING))
264264
handle_return(cuda.cuEventRecord(e, producer_s))
265265
handle_return(cuda.cuStreamWaitEvent(consumer_s, e, 0))
266+
handle_return(cuda.cuEventDestroy(e))
266267

267268
return buf
268269

@@ -275,6 +276,6 @@ def viewable(tuple arg_indices):
275276
cdef int idx
276277
for idx in arg_indices:
277278
args[idx] = _GPUMemoryViewProxy(args[idx])
278-
func(*args, **kwargs)
279+
return func(*args, **kwargs)
279280
return wrapped_func
280281
return wrapped_func_with_indices

0 commit comments

Comments
 (0)