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