3232import numpy
3333
3434import dpnp
35- import dpnp .dpnp_utils as utils
3635from dpnp .dpnp_array import dpnp_array
36+ from dpnp .dpnp_utils import get_usm_allocations , map_dtype_to_device
3737
3838__all__ = [
3939 "dpnp_geomspace" ,
@@ -60,7 +60,7 @@ def dpnp_geomspace(
6060 endpoint = True ,
6161 axis = 0 ,
6262):
63- usm_type_alloc , sycl_queue_alloc = utils . get_usm_allocations ([start , stop ])
63+ usm_type_alloc , sycl_queue_alloc = get_usm_allocations ([start , stop ])
6464
6565 if sycl_queue is None and device is None :
6666 sycl_queue = sycl_queue_alloc
@@ -77,7 +77,7 @@ def dpnp_geomspace(
7777 stop = _as_usm_ndarray (stop , _usm_type , sycl_queue_normalized )
7878
7979 dt = numpy .result_type (start , stop , float (num ))
80- dt = utils . map_dtype_to_device (dt , sycl_queue_normalized .sycl_device )
80+ dt = map_dtype_to_device (dt , sycl_queue_normalized .sycl_device )
8181 if dtype is None :
8282 dtype = dt
8383
@@ -144,7 +144,7 @@ def dpnp_linspace(
144144 retstep = False ,
145145 axis = 0 ,
146146):
147- usm_type_alloc , sycl_queue_alloc = utils . get_usm_allocations ([start , stop ])
147+ usm_type_alloc , sycl_queue_alloc = get_usm_allocations ([start , stop ])
148148
149149 if sycl_queue is None and device is None :
150150 sycl_queue = sycl_queue_alloc
@@ -164,7 +164,7 @@ def dpnp_linspace(
164164 stop = _as_usm_ndarray (stop , _usm_type , sycl_queue_normalized )
165165
166166 dt = numpy .result_type (start , stop , float (num ))
167- dt = utils . map_dtype_to_device (dt , sycl_queue_normalized .sycl_device )
167+ dt = map_dtype_to_device (dt , sycl_queue_normalized .sycl_device )
168168 if dtype is None :
169169 dtype = dt
170170
@@ -251,7 +251,7 @@ def dpnp_logspace(
251251 axis = 0 ,
252252):
253253 if not dpnp .isscalar (base ):
254- usm_type_alloc , sycl_queue_alloc = utils . get_usm_allocations (
254+ usm_type_alloc , sycl_queue_alloc = get_usm_allocations (
255255 [start , stop , base ]
256256 )
257257
0 commit comments