Replace dpctl.tensor with dpnp.tensor in dpnp

Author: vlad-perevezentsev

dpnp/dpnp_algo/dpnp_arraycreation.py

@@ -46,7 +46,7 @@
 
 
 def _as_usm_ndarray(a, usm_type, sycl_queue):
-    """Converts input object to `dpctl.tensor.usm_ndarray`"""
+    """Converts input object to `dpnp.tensor.usm_ndarray`"""
 
     if isinstance(a, dpnp_array):
         a = a.get_array()

dpnp/dpnp_algo/dpnp_elementwise_common.py

@@ -119,7 +119,7 @@ class DPNPUnaryFunc(UnaryElementwiseFunc):
             sycl_dev - The :class:`dpctl.SyclDevice` where the function
                 evaluation is carried out.
         The function is invoked when the argument of the unary function
-        requires casting, e.g. the argument of `dpctl.tensor.log` is an
+        requires casting, e.g. the argument of `dpnp.tensor.log` is an
         array with integral data type.
 
     """
@@ -137,7 +137,7 @@ def __init__(
         def _call_func(src, dst, sycl_queue, depends=None):
             """
             A callback to register in UnaryElementwiseFunc class of
-            dpctl.tensor
+            dpnp.tensor
             """
 
             if depends is None:
@@ -588,7 +588,7 @@ class DPNPBinaryFunc(BinaryElementwiseFunc):
                 evaluation is carried out.
         The function is only called when both arguments of the binary
         function require casting, e.g. both arguments of
-        `dpctl.tensor.logaddexp` are arrays with integral data type.
+        `dpnp.tensor.logaddexp` are arrays with integral data type.
     weak_type_resolver : {None, callable}, optional
         Function to influence type promotion behavior for Python scalar types
         of this binary function. The function takes 3 arguments:
@@ -615,7 +615,7 @@ def __init__(
         def _call_func(src1, src2, dst, sycl_queue, depends=None):
             """
             A callback to register in UnaryElementwiseFunc class of
-            dpctl.tensor
+            dpnp.tensor
             """
 
             if depends is None:

dpnp/dpnp_array.py

@@ -72,7 +72,7 @@ class dpnp_array:
     An array object represents a multidimensional tensor of numeric elements
     stored in a USM allocation on a SYCL device.
 
-    This is a wrapper around :class:`dpctl.tensor.usm_ndarray` that provides
+    This is a wrapper around :class:`dpnp.tensor.usm_ndarray` that provides
     methods to be compliant with original NumPy.
 
     """
@@ -609,12 +609,12 @@ def __usm_ndarray__(self):
         """
         Property to support ``__usm_ndarray__`` protocol.
 
-        It assumes to return :class:`dpctl.tensor.usm_ndarray` instance
+        It assumes to return :class:`dpnp.tensor.usm_ndarray` instance
         corresponding to the content of the object.
 
         This property is intended to speed-up conversion from
-        :class:`dpnp.ndarray` to :class:`dpctl.tensor.usm_ndarray` passed into
-        :func:`dpctl.tensor.asarray` function. The input object that implements
+        :class:`dpnp.ndarray` to :class:`dpnp.tensor.usm_ndarray` passed into
+        :func:`dpnp.tensor.asarray` function. The input object that implements
         ``__usm_ndarray__`` protocol is recognized as owner of USM allocation
         that is managed by a smart pointer, and asynchronous deallocation
         will not involve GIL.
@@ -631,13 +631,13 @@ def __xor__(self, other, /):
     def _create_from_usm_ndarray(usm_ary: dpt.usm_ndarray):
         """
         Return :class:`dpnp.ndarray` instance from USM allocation providing
-        by an instance of :class:`dpctl.tensor.usm_ndarray`.
+        by an instance of :class:`dpnp.tensor.usm_ndarray`.
 
         """
 
         if not isinstance(usm_ary, dpt.usm_ndarray):
             raise TypeError(
-                f"Expected dpctl.tensor.usm_ndarray, got {type(usm_ary)}"
+                f"Expected dpnp.tensor.usm_ndarray, got {type(usm_ary)}"
             )
         res = dpnp_array.__new__(dpnp_array)
         res._array_obj = usm_ary
@@ -826,7 +826,7 @@ def astype(
             `device` can be ``None``, a oneAPI filter selector string,
             an instance of :class:`dpctl.SyclDevice` corresponding to
             a non-partitioned SYCL device, an instance of
-            :class:`dpctl.SyclQueue`, or a :class:`dpctl.tensor.Device` object
+            :class:`dpctl.SyclQueue`, or a :class:`dpnp.tensor.Device` object
             returned by :attr:`dpnp.ndarray.device`.
             If the value is ``None``, returned array is created on the same
             device as that array.
@@ -937,7 +937,7 @@ def copy(
             `device` can be ``None``, a oneAPI filter selector string,
             an instance of :class:`dpctl.SyclDevice` corresponding to
             a non-partitioned SYCL device, an instance of
-            :class:`dpctl.SyclQueue`, or a :class:`dpctl.tensor.Device` object
+            :class:`dpctl.SyclQueue`, or a :class:`dpnp.tensor.Device` object
             returned by :attr:`dpnp.ndarray.device`.
 
             Default: ``None``.
@@ -1032,7 +1032,7 @@ def data(self):
     @property
     def device(self):
         """
-        Return :class:`dpctl.tensor.Device` object representing residence of
+        Return :class:`dpnp.tensor.Device` object representing residence of
         the array data.
 
         The ``Device`` object represents Array API notion of the device, and
@@ -1199,7 +1199,7 @@ def flatten(self, /, order="C"):
         return self.reshape(-1, order=order, copy=True)
 
     def get_array(self):
-        """Get :class:`dpctl.tensor.usm_ndarray` object."""
+        """Get :class:`dpnp.tensor.usm_ndarray` object."""
         return self._array_obj
 
     # 'getfield',
@@ -2052,7 +2052,7 @@ def to_device(self, device, /, *, stream=None):
             `device` can be ``None``, a oneAPI filter selector string,
             an instance of :class:`dpctl.SyclDevice` corresponding to
             a non-partitioned SYCL device, an instance of
-            :class:`dpctl.SyclQueue`, or a :class:`dpctl.tensor.Device` object
+            :class:`dpctl.SyclQueue`, or a :class:`dpnp.tensor.Device` object
             returned by :attr:`dpnp.ndarray.device`.
         stream : {SyclQueue, None}, optional
             Execution queue to synchronize with. If ``None``, synchronization

dpnp/dpnp_container.py

@@ -66,7 +66,7 @@ def arange(
     usm_type="device",
     sycl_queue=None,
 ):
-    """Validate input parameters before passing them into `dpctl.tensor` module"""
+    """Validate input parameters before passing them into `dpnp.tensor` module"""
     dpu.validate_usm_type(usm_type, allow_none=False)
     sycl_queue_normalized = dpnp.get_normalized_queue_device(
         sycl_queue=sycl_queue, device=device
@@ -153,7 +153,7 @@ def empty(
     usm_type="device",
     sycl_queue=None,
 ):
-    """Validate input parameters before passing them into `dpctl.tensor` module"""
+    """Validate input parameters before passing them into `dpnp.tensor` module"""
     dpu.validate_usm_type(usm_type, allow_none=False)
     sycl_queue_normalized = dpnp.get_normalized_queue_device(
         sycl_queue=sycl_queue, device=device
@@ -184,7 +184,7 @@ def eye(
     usm_type="device",
     sycl_queue=None,
 ):
-    """Validate input parameters before passing them into `dpctl.tensor` module"""
+    """Validate input parameters before passing them into `dpnp.tensor` module"""
     dpu.validate_usm_type(usm_type, allow_none=False)
     sycl_queue_normalized = dpnp.get_normalized_queue_device(
         sycl_queue=sycl_queue, device=device
@@ -215,7 +215,7 @@ def full(
     usm_type=None,
     sycl_queue=None,
 ):
-    """Validate input parameters before passing them into `dpctl.tensor` module"""
+    """Validate input parameters before passing them into `dpnp.tensor` module"""
     dpu.validate_usm_type(usm_type, allow_none=True)
 
     sycl_queue_normalized = dpnp.get_normalized_queue_device(
@@ -248,7 +248,7 @@ def ones(
     usm_type="device",
     sycl_queue=None,
 ):
-    """Validate input parameters before passing them into `dpctl.tensor` module"""
+    """Validate input parameters before passing them into `dpnp.tensor` module"""
     dpu.validate_usm_type(usm_type, allow_none=False)
     sycl_queue_normalized = dpnp.get_normalized_queue_device(
         sycl_queue=sycl_queue, device=device
@@ -288,7 +288,7 @@ def zeros(
     usm_type="device",
     sycl_queue=None,
 ):
-    """Validate input parameters before passing them into `dpctl.tensor` module"""
+    """Validate input parameters before passing them into `dpnp.tensor` module"""
     dpu.validate_usm_type(usm_type, allow_none=False)
     sycl_queue_normalized = dpnp.get_normalized_queue_device(
         sycl_queue=sycl_queue, device=device

dpnp/dpnp_iface.py

@@ -142,7 +142,7 @@ def asnumpy(a, order="C"):
 
 def as_usm_ndarray(a, dtype=None, device=None, usm_type=None, sycl_queue=None):
     """
-    Return :class:`dpctl.tensor.usm_ndarray` from input object `a`.
+    Return :class:`dpnp.tensor.usm_ndarray` from input object `a`.
 
     Parameters
     ----------
@@ -159,7 +159,7 @@ def as_usm_ndarray(a, dtype=None, device=None, usm_type=None, sycl_queue=None):
         `device` can be ``None``, a oneAPI filter selector string, an instance
         of :class:`dpctl.SyclDevice` corresponding to a non-partitioned SYCL
         device, an instance of :class:`dpctl.SyclQueue`, or a
-        :class:`dpctl.tensor.Device` object returned by
+        :class:`dpnp.tensor.Device` object returned by
         :attr:`dpnp.ndarray.device`.
         If the value is ``None``, returned array is created on the same device
         as `a`.
@@ -180,7 +180,7 @@ def as_usm_ndarray(a, dtype=None, device=None, usm_type=None, sycl_queue=None):
     out : usm_ndarray
         A dpctl USM ndarray from input array or scalar `a`.
         If `a` is instance of :class:`dpnp.ndarray`
-        or :class:`dpctl.tensor.usm_ndarray`, no array allocation will be done
+        or :class:`dpnp.tensor.usm_ndarray`, no array allocation will be done
         and `dtype`, `device`, `usm_type`, `sycl_queue` keywords
         will be ignored.
 
@@ -256,7 +256,7 @@ def check_limitations(
 def check_supported_arrays_type(*arrays, scalar_type=False, all_scalars=False):
     """
     Return ``True`` if each array has either type of scalar,
-    :class:`dpnp.ndarray` or :class:`dpctl.tensor.usm_ndarray`.
+    :class:`dpnp.ndarray` or :class:`dpnp.tensor.usm_ndarray`.
     But if any array has unsupported type, ``TypeError`` will be raised.
 
     Parameters
@@ -318,7 +318,7 @@ def default_float_type(device=None, sycl_queue=None):
         `device` can be ``None``, a oneAPI filter selector string, an instance
         of :class:`dpctl.SyclDevice` corresponding to a non-partitioned SYCL
         device, an instance of :class:`dpctl.SyclQueue`, or a
-        :class:`dpctl.tensor.Device` object returned by
+        :class:`dpnp.tensor.Device` object returned by
         :attr:`dpnp.ndarray.device`.
         The value ``None`` is interpreted as to use a default device.
 
@@ -434,7 +434,7 @@ def get_include():
 def get_normalized_queue_device(obj=None, device=None, sycl_queue=None):
     """
     Utility to process complementary keyword arguments 'device' and 'sycl_queue'
-    in subsequent calls of functions from `dpctl.tensor` module.
+    in subsequent calls of functions from `dpnp.tensor` module.
 
     If both arguments 'device' and 'sycl_queue' have default value ``None``
     and 'obj' has `sycl_queue` attribute, it assumes that Compute Follows Data
@@ -445,7 +445,7 @@ def get_normalized_queue_device(obj=None, device=None, sycl_queue=None):
     ----------
     obj : object, optional
         A python object. Can be an instance of `dpnp_array`,
-        `dpctl.tensor.usm_ndarray`, an object representing SYCL USM allocation
+        `dpnp.tensor.usm_ndarray`, an object representing SYCL USM allocation
         and implementing `__sycl_usm_array_interface__` protocol, an instance
         of `numpy.ndarray`, an object supporting Python buffer protocol,
         a Python scalar, or a (possibly nested) sequence of Python scalars.
@@ -462,7 +462,7 @@ def get_normalized_queue_device(obj=None, device=None, sycl_queue=None):
         `device` can be ``None``, a oneAPI filter selector string, an instance
         of :class:`dpctl.SyclDevice` corresponding to a non-partitioned SYCL
         device, an instance of :class:`dpctl.SyclQueue`, or a
-        :class:`dpctl.tensor.Device` object returned by
+        :class:`dpnp.tensor.Device` object returned by
         :attr:`dpnp.ndarray.device`.
         The value ``None`` is interpreted as to use the same device as `obj`.
 
@@ -472,7 +472,7 @@ def get_normalized_queue_device(obj=None, device=None, sycl_queue=None):
     -------
     sycl_queue: dpctl.SyclQueue
         A :class:`dpctl.SyclQueue` object normalized by
-        `normalize_queue_device` call of `dpctl.tensor` module invoked with
+        `normalize_queue_device` call of `dpnp.tensor` module invoked with
         `device` and `sycl_queue` values. If both incoming `device` and
         `sycl_queue` are ``None`` and `obj` has `sycl_queue` attribute,
         the normalization will be performed for `obj.sycl_queue` value.
@@ -540,13 +540,13 @@ def get_result_array(a, out=None, casting="safe"):
 
 def get_usm_ndarray(a):
     """
-    Return :class:`dpctl.tensor.usm_ndarray` from input array `a`.
+    Return :class:`dpnp.tensor.usm_ndarray` from input array `a`.
 
     Parameters
     ----------
     a : {dpnp.ndarray, usm_ndarray}
         Input array of supported type :class:`dpnp.ndarray`
-        or :class:`dpctl.tensor.usm_ndarray`.
+        or :class:`dpnp.tensor.usm_ndarray`.
 
     Returns
     -------
@@ -571,13 +571,13 @@ def get_usm_ndarray(a):
 
 def get_usm_ndarray_or_scalar(a):
     """
-    Return scalar or :class:`dpctl.tensor.usm_ndarray` from input object `a`.
+    Return scalar or :class:`dpnp.tensor.usm_ndarray` from input object `a`.
 
     Parameters
     ----------
     a : {scalar, dpnp_array, usm_ndarray}
         Input of any supported type: scalar, :class:`dpnp.ndarray`
-        or :class:`dpctl.tensor.usm_ndarray`.
+        or :class:`dpnp.tensor.usm_ndarray`.
 
     Returns
     -------
@@ -634,7 +634,7 @@ def is_cuda_backend(obj=None):
 def is_supported_array_or_scalar(a):
     """
     Return ``True`` if `a` is a scalar or an array of either
-    :class:`dpnp.ndarray` or :class:`dpctl.tensor.usm_ndarray` type,
+    :class:`dpnp.ndarray` or :class:`dpnp.tensor.usm_ndarray` type,
     ``False`` otherwise.
 
     Parameters
@@ -656,7 +656,7 @@ def is_supported_array_or_scalar(a):
 def is_supported_array_type(a):
     """
     Return ``True`` if an array of either type :class:`dpnp.ndarray`
-    or :class:`dpctl.tensor.usm_ndarray` type, ``False`` otherwise.
+    or :class:`dpnp.tensor.usm_ndarray` type, ``False`` otherwise.
 
     Parameters
     ----------