Implement check_zeros_shape_gesvd

Author: vlad-perevezentsev

dpnp/backend/extensions/lapack/gesvd.cpp

@@ -158,7 +158,14 @@ std::pair<sycl::event, sycl::event>
                                      jobu_val, jobvt_val, expected_a_u_vt_ndim,
                                      expected_s_ndim);
 
-    // TODO: check non_zero shape
+    // Ensure `m` and 'n' are non-zero, otherwise return empty
+    // events
+    if (gesvd_utils::check_zeros_shape_gesvd(a_array, out_s, out_u, out_vt,
+                                             jobu_val, jobvt_val))
+    {
+        // nothing to do
+        return std::make_pair(sycl::event(), sycl::event());
+    }
 
     auto array_types = dpctl_td_ns::usm_ndarray_types();
     int a_array_type_id =

dpnp/backend/extensions/lapack/gesvd_batch.cpp

@@ -222,7 +222,14 @@ std::pair<sycl::event, sycl::event>
                                      jobu_val, jobvt_val, expected_a_u_vt_ndim,
                                      expected_s_ndim);
 
-    // TODO: check non_zero shape
+    // Ensure `batch_size`, `m` and 'n' are non-zero, otherwise return empty
+    // events
+    if (gesvd_utils::check_zeros_shape_gesvd(a_array, out_s, out_u, out_vt,
+                                             jobu_val, jobvt_val))
+    {
+        // nothing to do
+        return std::make_pair(sycl::event(), sycl::event());
+    }
 
     auto array_types = dpctl_td_ns::usm_ndarray_types();
     int a_array_type_id =

dpnp/backend/extensions/lapack/gesvd_common_utils.hpp

@@ -32,6 +32,8 @@
 #include "utils/output_validation.hpp"
 #include "utils/type_dispatch.hpp"
 
+#include "common_helpers.hpp"
+
 namespace dpnp::extensions::lapack::gesvd_utils
 {
 namespace dpctl_td_ns = dpctl::tensor::type_dispatch;
@@ -170,6 +172,41 @@ inline void common_gesvd_checks(sycl::queue &exec_q,
     }
 }
 
+// Checks if the shape of input arrays for gesvd has any non-zero dimension.
+inline bool check_zeros_shape_gesvd(dpctl::tensor::usm_ndarray a_array,
+                                    dpctl::tensor::usm_ndarray out_s,
+                                    dpctl::tensor::usm_ndarray out_u,
+                                    dpctl::tensor::usm_ndarray out_vt,
+                                    const std::int8_t jobu_val,
+                                    const std::int8_t jobvt_val)
+{
+
+    const int a_array_nd = a_array.get_ndim();
+    const int out_u_array_nd = out_u.get_ndim();
+    const int out_s_array_nd = out_s.get_ndim();
+    const int out_vt_array_nd = out_vt.get_ndim();
+
+    const py::ssize_t *a_array_shape = a_array.get_shape_raw();
+    const py::ssize_t *s_out_shape = out_s.get_shape_raw();
+    const py::ssize_t *u_out_shape = out_u.get_shape_raw();
+    const py::ssize_t *vt_out_shape = out_vt.get_shape_raw();
+
+    bool is_zeros_shape = helper::check_zeros_shape(a_array_nd, a_array_shape);
+    if (jobu_val == 'N' && jobvt_val == 'N') {
+        is_zeros_shape = is_zeros_shape || helper::check_zeros_shape(
+                                               out_vt_array_nd, vt_out_shape);
+    }
+    else {
+        is_zeros_shape =
+            is_zeros_shape ||
+            helper::check_zeros_shape(out_u_array_nd, s_out_shape) ||
+            helper::check_zeros_shape(out_s_array_nd, u_out_shape) ||
+            helper::check_zeros_shape(out_vt_array_nd, vt_out_shape);
+    }
+
+    return is_zeros_shape;
+}
+
 inline void handle_lapack_exc(std::int64_t scratchpad_size,
                               const oneapi::mkl::lapack::exception &e,
                               std::stringstream &error_msg)