fix viewable return

NVIDIA · Sep 6, 2024 · 9f902b1 · 9f902b1
1 parent 60682de
commit 9f902b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cuda_py/cuda/py/_memoryview.pyx b/cuda_py/cuda/py/_memoryview.pyx
@@ -275,6 +275,6 @@ def viewable(tuple arg_indices):
             cdef int idx
             for idx in arg_indices:
                 args[idx] = _GPUMemoryViewProxy(args[idx])
-            func(*args, **kwargs)
+            return func(*args, **kwargs)
         return wrapped_func
     return wrapped_func_with_indices