Skip to content

Commit 1ee1e91

Browse files
Memory objects with queue=None get queue from cache
Queue is looked up from cache for default-selected device.
1 parent 81b362d commit 1ee1e91

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dpctl/memory/_memory.pyx

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ from dpctl._backend cimport ( # noqa: E211
6161
from .._sycl_context cimport SyclContext
6262
from .._sycl_device cimport SyclDevice
6363
from .._sycl_queue cimport SyclQueue
64+
from .._sycl_queue_manager cimport get_device_cached_queue
6465

6566
import collections
6667
import numbers
@@ -150,7 +151,7 @@ cdef class _Memory:
150151

151152
if (nbytes > 0):
152153
if queue is None:
153-
queue = dpctl.SyclQueue()
154+
queue = get_device_cached_queue(dpctl.SyclDevice())
154155

155156
QRef = queue.get_queue_ref()
156157
if (ptr_type == b"shared"):

0 commit comments

Comments
 (0)