Attempt 2 at custom-dtype arguments

inducer · May 27, 2014 · df49c03 · df49c03
1 parent 9cd4203
commit df49c03
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pycuda/driver.py b/pycuda/driver.py
@@ -116,7 +116,7 @@ def _build_arg_buf(args):
         arg_data = []
         format = ""
         for i, arg in enumerate(args):
-            if isinstance(arg, (np.number, np.void)):
+            if isinstance(arg, np.number):
                 arg_data.append(arg)
                 format += arg.dtype.char
             elif isinstance(arg, (DeviceAllocation, PooledDeviceAllocation)):
@@ -133,6 +133,9 @@ def _build_arg_buf(args):
                 else:
                     arg_data.append(arg)
                     format += "%ds" % arg.nbytes
+            elif isinstance(arg, np.void):
+                arg_data.append(str(buffer(arg)))
+                format += "%ds" % arg.itemsize
             else:
                 try:
                     gpudata = np.intp(arg.gpudata)