Skip to content

Commit bd46f72

Browse files
committed
Use data model from context in dpnp_iface
1 parent e615596 commit bd46f72

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

numba_dpex/dpnp_iface/_intrinsic.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
populate_array,
2323
)
2424

25-
from numba_dpex.core.datamodel.models import dpex_data_model_manager as dpex_dmm
2625
from numba_dpex.core.runtime import context as dpexrt
2726
from numba_dpex.core.types import DpnpNdArray
2827
from numba_dpex.core.types.dpctl_types import DpctlSyclQueue
@@ -142,13 +141,15 @@ def _get_queue_ref(
142141
raise AssertionError(
143142
"Expected the queue_arg to be an llvmir.LiteralStructType"
144143
)
145-
sycl_queue_dm = dpex_dmm.lookup(sycl_queue_arg.numba_ty)
144+
sycl_queue_dm = context.data_model_manager.lookup(
145+
sycl_queue_arg.numba_ty
146+
)
146147
queue_ref = builder.extract_value(
147148
sycl_queue_arg.llvmir_val,
148149
sycl_queue_dm.get_field_position("queue_ref"),
149150
)
150151
elif array_arg is not None:
151-
dpnp_ndarray_dm = dpex_dmm.lookup(array_arg.numba_ty)
152+
dpnp_ndarray_dm = context.data_model_manager.lookup(array_arg.numba_ty)
152153
queue_ref = builder.extract_value(
153154
array_arg.llvmir_val,
154155
dpnp_ndarray_dm.get_field_position("sycl_queue"),

numba_dpex/dpnp_iface/arrayobj.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
from numba.np.arrayobj import getitem_arraynd_intp as np_getitem_arraynd_intp
1616
from numba.np.numpy_support import is_nonelike
1717

18-
from numba_dpex.core.datamodel.models import dpex_data_model_manager as dpex_dmm
1918
from numba_dpex.core.types import DpnpNdArray
2019

2120
from ._intrinsic import (
@@ -1083,9 +1082,9 @@ def getitem_arraynd_intp(context, builder, sig, args):
10831082
if isinstance(sig.return_type, DpnpNdArray):
10841083
array_val = args[0]
10851084
array_ty = sig.args[0]
1086-
sycl_queue_attr_pos = dpex_dmm.lookup(array_ty).get_field_position(
1087-
"sycl_queue"
1088-
)
1085+
sycl_queue_attr_pos = context.data_model_manager.lookup(
1086+
array_ty
1087+
).get_field_position("sycl_queue")
10891088
sycl_queue_attr = builder.extract_value(array_val, sycl_queue_attr_pos)
10901089
ret = builder.insert_value(ret, sycl_queue_attr, sycl_queue_attr_pos)
10911090

0 commit comments

Comments
 (0)