File tree Expand file tree Collapse file tree 1 file changed +3
-6
lines changed Expand file tree Collapse file tree 1 file changed +3
-6
lines changed Original file line number Diff line number Diff line change 14
14
# See the License for the specific language governing permissions and
15
15
# limitations under the License.
16
16
import dpctl
17
+ from dpctl ._sycl_queue_manager import get_device_cached_queue
17
18
18
19
__doc__ = "Implementation of array API mandated Device class"
19
20
@@ -60,9 +61,7 @@ def create_device(cls, dev):
60
61
elif isinstance (dev , dpctl .SyclDevice ):
61
62
par = dev .parent_device
62
63
if par is None :
63
- if dev not in cls .__device_queue_map__ :
64
- cls .__device_queue_map__ [dev ] = dpctl .SyclQueue (dev )
65
- obj .sycl_queue_ = cls .__device_queue_map__ [dev ]
64
+ obj .sycl_queue_ = get_device_cached_queue (dev )
66
65
else :
67
66
raise ValueError (
68
67
f"Using non-root device { dev } to specify offloading "
@@ -74,9 +73,7 @@ def create_device(cls, dev):
74
73
_dev = dpctl .SyclDevice ()
75
74
else :
76
75
_dev = dpctl .SyclDevice (dev )
77
- if _dev not in cls .__device_queue_map__ :
78
- cls .__device_queue_map__ [_dev ] = dpctl .SyclQueue (_dev )
79
- obj .sycl_queue_ = cls .__device_queue_map__ [_dev ]
76
+ obj .sycl_queue_ = get_device_cached_queue (_dev )
80
77
return obj
81
78
82
79
@property
You can’t perform that action at this time.
0 commit comments