Allow a managed numpy array to be passed directly to a kernel as if it

was a gpuarray.
inducer · Feb 17, 2014 · 817928c · 817928c
1 parent 55e4acb
commit 817928c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pycuda/driver.py b/pycuda/driver.py
@@ -117,8 +117,12 @@ def _build_arg_buf(args):
                 arg_data.append(int(arg.get_device_alloc()))
                 format += "P"
             elif isinstance(arg, np.ndarray):
-                arg_data.append(arg)
-                format += "%ds" % arg.nbytes
+                if isinstance(arg.base, (ManagedAllocation,)):
+                    arg_data.append(int(arg.base))
+                    format += "P"
+                else:
+                    arg_data.append(arg)
+                    format += "%ds" % arg.nbytes
             else:
                 try:
                     gpudata = np.intp(arg.gpudata)