Accept tuple value for out keyword passing to ufuncs

Author: antonwolfy

dpnp/dpnp_algo/dpnp_elementwise_common.py

@@ -199,9 +199,15 @@ def __call__(
         if dtype is not None:
             x_usm = dpt.astype(x_usm, dtype, copy=False)
 
+        if isinstance(out, tuple):
+            if len(out) != self.nout:
+                raise ValueError(
+                    "'out' tuple must have exactly one entry per ufunc output"
+                )
+            out = out[0]
         out_usm = None if out is None else dpnp.get_usm_ndarray(out)
-        res_usm = super().__call__(x_usm, out=out_usm, order=order)
 
+        res_usm = super().__call__(x_usm, out=out_usm, order=order)
         if out is not None and isinstance(out, dpnp_array):
             return out
         return dpnp_array._create_from_usm_ndarray(res_usm)
@@ -361,7 +367,7 @@ def __call__(
         orig_out, out = list(out), list(out)
         res_dts = [res1_dt, res2_dt]
 
-        for i in range(2):
+        for i in range(self.nout):
             if out[i] is None:
                 continue
 
@@ -419,7 +425,7 @@ def __call__(
             dep_evs = copy_ev
 
         # Allocate a buffer for the output arrays if needed
-        for i in range(2):
+        for i in range(self.nout):
             if out[i] is None:
                 res_dt = res_dts[i]
                 if order == "K":
@@ -438,7 +444,7 @@ def __call__(
         )
         _manager.add_event_pair(ht_unary_ev, unary_ev)
 
-        for i in range(2):
+        for i in range(self.nout):
             orig_res, res = orig_out[i], out[i]
             if not (orig_res is None or orig_res is res):
                 # Copy the out data from temporary buffer to original memory
@@ -606,6 +612,13 @@ def __call__(
 
         x1_usm = dpnp.get_usm_ndarray_or_scalar(x1)
         x2_usm = dpnp.get_usm_ndarray_or_scalar(x2)
+
+        if isinstance(out, tuple):
+            if len(out) != self.nout:
+                raise ValueError(
+                    "'out' tuple must have exactly one entry per ufunc output"
+                )
+            out = out[0]
         out_usm = None if out is None else dpnp.get_usm_ndarray(out)
 
         if (