Use function from dpctl_ext.tensor in tensor python files

Author: vlad-perevezentsev

dpctl_ext/tensor/_ctors.py

@@ -360,7 +360,7 @@ def _copy_through_host_walker(seq_o, usm_res):
             )
             is None
         ):
-            usm_res[...] = dpt.asnumpy(seq_o).copy()
+            usm_res[...] = dpt_ext.asnumpy(seq_o).copy()
             return
         else:
             usm_res[...] = seq_o
@@ -1440,7 +1440,7 @@ def linspace(
     )
     _manager.add_event_pair(hev, la_ev)
 
-    return res if int_dt is None else dpt.astype(res, int_dt)
+    return res if int_dt is None else dpt_ext.astype(res, int_dt)
 
 
 def meshgrid(*arrays, indexing="xy"):

dpctl_ext/tensor/_indexing_functions.py

@@ -190,7 +190,7 @@ def place(arr, mask, vals):
     if vals.dtype == arr.dtype:
         rhs = vals
     else:
-        rhs = dpt.astype(vals, arr.dtype)
+        rhs = dpt_ext.astype(vals, arr.dtype)
     hev, pl_ev = ti._place(
         dst=arr,
         cumsum=cumsum,

dpctl_ext/tensor/_reduction.py

@@ -506,7 +506,7 @@ def count_nonzero(x, /, *, axis=None, keepdims=False, out=None):
             type.
     """
     if x.dtype != dpt.bool:
-        x = dpt.astype(x, dpt.bool, copy=False)
+        x = dpt_ext.astype(x, dpt.bool, copy=False)
     return sum(
         x,
         axis=axis,

dpctl_ext/tensor/_utility_functions.py

@@ -489,7 +489,7 @@ def diff(x, /, *, axis=-1, n=1, prepend=None, append=None):
         slice(None) if i != axis else slice(None, -1) for i in range(x_nd)
     )
 
-    diff_op = dpt.not_equal if x.dtype == dpt.bool else dpt.subtract
+    diff_op = dpt_ext.not_equal if x.dtype == dpt.bool else dpt_ext.subtract
     if n > 1:
         arr_tmp0 = diff_op(arr[sl0], arr[sl1])
         arr_tmp1 = diff_op(arr_tmp0[sl0], arr_tmp0[sl1])