Attempt 4 at custom-dtype arguments

inducer · May 27, 2014 · 9a0ff04 · 9a0ff04
1 parent 570d664
commit 9a0ff04
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pycuda/driver.py b/pycuda/driver.py
@@ -14,8 +14,10 @@
 
 if sys.version_info >= (3,):
     _memoryview = memoryview
+    _my_bytes = bytes
 else:
     _memoryview = buffer
+    _my_bytes = str
 
 
 try:
@@ -141,7 +143,7 @@ def _build_arg_buf(args):
                     arg_data.append(arg)
                     format += "%ds" % arg.nbytes
             elif isinstance(arg, np.void):
-                arg_data.append(str(_memoryview(arg)))
+                arg_data.append(_my_bytes(_memoryview(arg)))
                 format += "%ds" % arg.itemsize
             else:
                 try: