From f6360a2d7fb33e0bd0f065c774112ba6b26246d5 Mon Sep 17 00:00:00 2001
From: Matt Wala <wala1@illinois.edu>
Date: Fri, 23 Jun 2017 00:36:41 -0500
Subject: [PATCH] Array.copy(): Change to update the queue of returned array.

---
 pyopencl/array.py | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/pyopencl/array.py b/pyopencl/array.py
index 08f7cc8b..adbb7aad 100644
--- a/pyopencl/array.py
+++ b/pyopencl/array.py
@@ -631,14 +631,24 @@ class Array(object):
 
         return ary
 
-    def copy(self, queue=None):
-        """.. versionadded:: 2013.1"""
+    def copy(self, queue=_copy_queue):
+        """
+        :arg queue: The :class:`CommandQueue` for the returned array.
 
-        queue = queue or self.queue
-        result = self._new_like_me()
+        .. versionchanged:: 2017.1.2
+            Updates the queue of returned array.
+
+        .. versionadded:: 2013.1
+        """
+
+        if queue is _copy_queue:
+            queue = self.queue
+
+        result = self._new_like_me(queue=queue)
 
         if self.nbytes:
-            cl.enqueue_copy(queue, result.base_data, self.base_data,
+            cl.enqueue_copy(queue or self.queue,
+                    result.base_data, self.base_data,
                     src_offset=self.offset, byte_count=self.nbytes)
 
         return result
-- 
GitLab