We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4bf151c commit 93c0418Copy full SHA for 93c0418
numba_dpex/core/types/usm_ndarray_type.py
@@ -59,7 +59,9 @@ def __init__(
59
"The device keyword arg should be a str object specifying "
60
"a SYCL filter selector"
61
)
62
- self.queue = dpctl.SyclQueue(device)
+ self.queue = dpctl.tensor._device.normalize_queue_device(
63
+ device=device
64
+ )
65
self.device = device
66
elif queue is not None and device == "unknown":
67
if not isinstance(queue, dpctl.SyclQueue):
@@ -69,7 +71,7 @@ def __init__(
69
71
self.device = self.queue.sycl_device.filter_string
70
72
self.queue = queue
73
else:
- self.queue = dpctl.SyclQueue()
74
+ self.queue = dpctl.tensor._device.normalize_queue_device()
75
76
77
if not dtype:
0 commit comments