Fix gpuarray.set_async

inducer · Jul 29, 2015 · 51599aa · 51599aa
1 parent 087a97b
commit 51599aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycuda/gpuarray.py b/pycuda/gpuarray.py
@@ -1161,7 +1161,7 @@ def _memcpy_discontig(dst, src, async=False, stream=None):
         else:
             src = _as_strided(src, shape=(src.size,), strides=(src.dtype.itemsize,))
             if async:
-                drv.memcpy_htod(dst.gpudata, src, stream=stream)
+                drv.memcpy_htod_async(dst.gpudata, src, stream=stream)
             else:
                 drv.memcpy_htod(dst.gpudata, src)
         return