Fix issue gh-2311 (#2314)

This PR suggests a solving #2311

When the input array is `(n, m)` or `(batch_size, n, m)` where m=1
`geqrf`/`geqrf_batch` call returns F-contig result array and
`dpnp.empty_like()` is used to allocate `q` allocates F-contig array and
this raises `ValueError` exception in `orgqr` and `ungqr` functions.

Added `q` allocation always C-contig and updated tests

Author: vlad-perevezentsev

dpnp/linalg/dpnp_utils_linalg.py

@@ -449,13 +449,15 @@ def _batched_qr(a, mode="reduced"):
             a_t,
             shape=(batch_size, m, m),
             dtype=res_type,
+            order="C",
         )
     else:
         mc = k
         q = dpnp.empty_like(
             a_t,
             shape=(batch_size, n, m),
             dtype=res_type,
+            order="C",
         )
 
     # use DPCTL tensor function to fill the matrix array `q[..., :n, :]`
@@ -2532,13 +2534,15 @@ def dpnp_qr(a, mode="reduced"):
             a_t,
             shape=(m, m),
             dtype=res_type,
+            order="C",
         )
     else:
         mc = k
         q = dpnp.empty_like(
             a_t,
             shape=(n, m),
             dtype=res_type,
+            order="C",
         )
 
     # use DPCTL tensor function to fill the matrix array `q[:n]`

dpnp/tests/test_linalg.py

@@ -2372,12 +2372,24 @@ class TestQr:
     @pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))
     @pytest.mark.parametrize(
         "shape",
-        [(2, 2), (3, 4), (5, 3), (16, 16), (2, 2, 2), (2, 4, 2), (2, 2, 4)],
+        [
+            (2, 1),
+            (2, 2),
+            (3, 4),
+            (5, 3),
+            (16, 16),
+            (3, 3, 1),
+            (2, 2, 2),
+            (2, 4, 2),
+            (2, 2, 4),
+        ],
         ids=[
+            "(2, 1)",
             "(2, 2)",
             "(3, 4)",
             "(5, 3)",
             "(16, 16)",
+            "(3, 3, 1)",
             "(2, 2, 2)",
             "(2, 4, 2)",
             "(2, 2, 4)",