Skip to content

Commit 8daca71

Browse files
get_device_cached_queue now supports filter selector strings
``` In [1]: import dpctl In [2]: import dpctl._sycl_queue_manager as qm In [3]: qm.__pyx_capi__ Out[3]: {'get_current_queue': <capsule object "struct PySyclQueueObject *(int __pyx_skip_dispatch)" at 0x7fe2260bcc30>, 'get_current_device_type': <capsule object "PyObject *(int __pyx_skip_dispatch)" at 0x7fe2260bcc60>, 'get_current_backend': <capsule object "PyObject *(int __pyx_skip_dispatch)" at 0x7fe2260bcc90>, 'get_device_cached_queue': <capsule object "PyObject *(PyObject *, int __pyx_skip_dispatch)" at 0x7fe2260bccc0>} In [4]: qm.get_device_cached_queue("cpu") Out[4]: <dpctl.SyclQueue at 0x7fe226945b80> In [5]: qm.get_device_cached_queue("gpu") Out[5]: <dpctl.SyclQueue at 0x7fe21c032ec0> ```
1 parent ec19e60 commit 8daca71

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

dpctl/_sycl_queue_manager.pyx

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -310,6 +310,9 @@ cdef class _DeviceDefaultQueueCache:
310310
elif isinstance(key, SyclDevice):
311311
q = SyclQueue(key)
312312
ctx_dev = q.sycl_context, key
313+
elif isinstance(key, str):
314+
q = SyclQueue(key)
315+
ctx_dev = q.sycl_context, q.sycl_device
313316
else:
314317
raise TypeError
315318
if ctx_dev in self.__device_queue_map__:

0 commit comments

Comments
 (0)