Add sycl_queue and usm_type tests

Author: vlad-perevezentsev

dpnp/linalg/dpnp_utils_linalg.py

@@ -2303,11 +2303,15 @@ def dpnp_lu_factor(a, overwrite_a=False, check_finite=True):
     """
 
     res_type = _common_type(a)
+    a_sycl_queue = a.sycl_queue
+    a_usm_type = a.usm_type
 
     # accommodate empty arrays
     if a.size == 0:
         lu = dpnp.empty_like(a)
-        piv = dpnp.arange(0, dtype=dpnp.int64)
+        piv = dpnp.arange(
+            0, dtype=dpnp.int64, usm_type=a_usm_type, sycl_queue=a_sycl_queue
+        )
         return lu, piv
 
     if check_finite:
@@ -2317,12 +2321,7 @@ def dpnp_lu_factor(a, overwrite_a=False, check_finite=True):
     if a.ndim > 2:
         raise NotImplementedError("Batched matrices are not supported")
 
-    m, n = a.shape
-
-    a_sycl_queue = a.sycl_queue
-    a_usm_type = a.usm_type
     _manager = dpu.SequentialOrderManager[a_sycl_queue]
-
     a_usm_arr = dpnp.get_usm_ndarray(a)
 
     # SciPy-compatible behavior
@@ -2345,6 +2344,8 @@ def dpnp_lu_factor(a, overwrite_a=False, check_finite=True):
         a_h = a
         copy_ev = None
 
+    m, n = a.shape
+
     ipiv_h = dpnp.empty(
         min(m, n),
         dtype=dpnp.int64,

dpnp/tests/test_sycl_queue.py

@@ -1570,6 +1570,18 @@ def test_lstsq(self, m, n, nrhs, device):
             assert_sycl_queue_equal(param_queue, a.sycl_queue)
             assert_sycl_queue_equal(param_queue, b.sycl_queue)
 
+    @pytest.mark.parametrize(
+        "data",
+        [[[1.0, 2.0], [3.0, 5.0]], [[]]],
+    )
+    def test_lu_factor(self, data, device):
+        a = dpnp.array(data, device=device)
+        result = dpnp.linalg.lu_factor(a)
+
+        for param in result:
+            param_queue = param.sycl_queue
+            assert_sycl_queue_equal(param_queue, a.sycl_queue)
+
     @pytest.mark.parametrize("n", [-1, 0, 1, 2, 3])
     def test_matrix_power(self, n, device):
         x = dpnp.array([[1.0, 2.0], [3.0, 5.0]], device=device)

dpnp/tests/test_usm_type.py

@@ -1449,6 +1449,18 @@ def test_lstsq(self, m, n, nrhs, usm_type, usm_type_other):
                 [usm_type, usm_type_other]
             )
 
+    @pytest.mark.parametrize(
+        "data",
+        [[[1.0, 2.0], [3.0, 5.0]], [[]]],
+    )
+    def test_lu_factor(self, data, usm_type):
+        a = dpnp.array(data, usm_type=usm_type)
+        result = dpnp.linalg.lu_factor(a)
+
+        assert a.usm_type == usm_type
+        for param in result:
+            assert param.usm_type == a.usm_type
+
     @pytest.mark.parametrize("n", [-1, 0, 1, 2, 3])
     def test_matrix_power(self, n, usm_type):
         a = dpnp.array([[1, 2], [3, 5]], usm_type=usm_type)