implement dpnp.einsum

Author: vtavana

.github/workflows/conda-package.yml

@@ -39,6 +39,7 @@ env:
       third_party/cupy/linalg_tests/test_norms.py
       third_party/cupy/linalg_tests/test_product.py
       third_party/cupy/linalg_tests/test_solve.py
+      third_party/cupy/linalg_tests/test_einsum.py
       third_party/cupy/logic_tests/test_comparison.py
       third_party/cupy/logic_tests/test_truth.py
       third_party/cupy/manipulation_tests/test_basic.py

dpnp/dpnp_array.py

@@ -549,12 +549,14 @@ def astype(self, dtype, order="K", casting="unsafe", subok=True, copy=True):
             If it is False and no cast happens, then this method returns the array itself.
             Otherwise, a copy is returned.
         casting : {'no', 'equiv', 'safe', 'same_kind', 'unsafe'}, optional
-            Controls what kind of data casting may occur. Defaults to 'unsafe' for backwards compatibility.
-            'no' means the data types should not be cast at all.
-            'equiv' means only byte-order changes are allowed.
-            'safe' means only casts which can preserve values are allowed.
-            'same_kind' means only safe casts or casts within a kind, like float64 to float32, are allowed.
-            'unsafe' means any data conversions may be done.
+            Controls what kind of data casting may occur.
+            Defaults to ``'unsafe'`` for backwards compatibility.
+            - 'no' means the data types should not be cast at all.
+            - 'equiv' means only byte-order changes are allowed.
+            - 'safe' means only casts which can preserve values are allowed.
+            - 'same_kind' means only safe casts or casts within a kind, like
+              float64 to float32, are allowed.
+            - 'unsafe' means any data conversions may be done.
         copy : bool, optional
             By default, astype always returns a newly allocated array. If this is set to false, and the dtype,
             order, and subok requirements are satisfied, the input array is returned instead of a copy.

dpnp/dpnp_iface.py

@@ -202,7 +202,7 @@ def astype(x1, dtype, order="K", casting="unsafe", copy=True):
             - 'equiv' means only byte-order changes are allowed.
             - 'safe' means only casts which can preserve values are allowed.
             - 'same_kind' means only safe casts or casts within a kind, like
-            float64 to float32, are allowed.
+              float64 to float32, are allowed.
             - 'unsafe' means any data conversions may be done.
     copy : bool, optional
         By default, astype always returns a newly allocated array. If this
@@ -326,7 +326,7 @@ def check_supported_arrays_type(*arrays, scalar_type=False, all_scalars=False):
             f"An array must be any of supported type, but got {type(a)}"
         )
 
-    if len(arrays) > 1 and not (all_scalars or any_is_array):
+    if len(arrays) > 0 and not (all_scalars or any_is_array):
         raise TypeError(
             "At least one input must be of supported array type, "
             "but got all scalars."