Using cached queue instead of creating new one on type inference

IntelPython · Mar 2, 2023 · 8f23f93 · 8f23f93
1 parent 4bf151c
commit 8f23f93
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/numba_dpex/core/types/usm_ndarray_type.py b/numba_dpex/core/types/usm_ndarray_type.py
@@ -59,7 +59,7 @@ def __init__(
                     "The device keyword arg should be a str object specifying "
                     "a SYCL filter selector"
                 )
-            self.queue = dpctl.SyclQueue(device)
+            self.queue = dpctl.tensor._device.normalize_queue_device(device=device)
             self.device = device
         elif queue is not None and device == "unknown":
             if not isinstance(queue, dpctl.SyclQueue):
@@ -69,7 +69,7 @@ def __init__(
             self.device = self.queue.sycl_device.filter_string
             self.queue = queue
         else:
-            self.queue = dpctl.SyclQueue()
+            self.queue = dpctl.tensor._device.normalize_queue_device()
             self.device = self.queue.sycl_device.filter_string
 
         if not dtype: