Merge pull request #83 from sjperkins/fix-set-async-stream-param

Fix stream parameters in asynchronous methods.
inducer · Aug 7, 2015 · c8c7a6a · c8c7a6a
2 parents dae67bf + 5d5e5a2
commit c8c7a6a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pycuda/gpuarray.py b/pycuda/gpuarray.py
@@ -236,7 +236,7 @@ def set(self, ary, async=False, stream=None):
             _memcpy_discontig(self, ary, async=async, stream=stream)
 
     def set_async(self, ary, stream=None):
-        return self.set(ary, async=True, stream=None)
+        return self.set(ary, async=True, stream=stream)
 
     def get(self, ary=None, pagelocked=False, async=False, stream=None):
         if ary is None:
@@ -1191,7 +1191,7 @@ def _memcpy_discontig(dst, src, async=False, stream=None):
 
     if len(shape) == 2:
         if async:
-            copy(stream=stream)
+            copy(stream)
         else:
             copy(aligned=True)
 
@@ -1206,7 +1206,7 @@ def _memcpy_discontig(dst, src, async=False, stream=None):
 
         copy.depth = shape[2]
         if async:
-            copy(stream=stream)
+            copy(stream)
         else:
             copy()