diff --git a/python/rmm/_lib/device_buffer.pyx b/python/rmm/_lib/device_buffer.pyx index bc761c29c..d33198315 100644 --- a/python/rmm/_lib/device_buffer.pyx +++ b/python/rmm/_lib/device_buffer.pyx @@ -344,7 +344,7 @@ cdef void _copy_async(const void* src, void* dst, size_t count, ccudart.cudaMemcpyKind kind, - cuda_stream_view stream) nogil: + cuda_stream_view stream) nogil except *: """ Asynchronously copy data between host and/or device pointers