File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change 5
5
from dpctl import SyclQueue
6
6
from dpctl .tensor import usm_ndarray
7
7
from dpnp import ndarray
8
- from numba .core import types
9
8
from numba .extending import typeof_impl
10
9
from numba .np import numpy_support
11
10
16
15
from ..types .usm_ndarray_type import USMNdArray
17
16
18
17
19
- def _typeof_helper (val , array_class_type ):
18
+ def _array_typeof_helper (val , array_class_type ):
20
19
"""Creates a Numba type of the specified ``array_class_type`` for ``val``."""
21
20
try :
22
21
dtype = numpy_support .from_dtype (val .dtype )
@@ -74,7 +73,7 @@ def typeof_usm_ndarray(val, c):
74
73
75
74
Returns: The Numba type corresponding to dpctl.tensor.usm_ndarray
76
75
"""
77
- return _typeof_helper (val , USMNdArray )
76
+ return _array_typeof_helper (val , USMNdArray )
78
77
79
78
80
79
@typeof_impl .register (ndarray )
@@ -92,7 +91,7 @@ def typeof_dpnp_ndarray(val, c):
92
91
93
92
Returns: The Numba type corresponding to dpnp.ndarray
94
93
"""
95
- return _typeof_helper (val , DpnpNdArray )
94
+ return _array_typeof_helper (val , DpnpNdArray )
96
95
97
96
98
97
@typeof_impl .register (SyclQueue )
You can’t perform that action at this time.
0 commit comments