Skip tests on arl or mtl

Author: vlad-perevezentsev

dpnp/tests/helper.py

@@ -404,6 +404,14 @@ def has_support_aspect64(device=None):
     return dev.has_aspect_fp64
 
 
+def is_arl_or_mtl(device=None):
+    """
+    Return True if a test is running on Arrow Lake or Meteor Lake GPU device,
+    False otherwise.
+    """
+    return _get_dev_mask(device) == 0x7D00
+
+
 def is_cpu_device(device=None):
     """
     Return True if a test is running on CPU device, False otherwise.
@@ -464,14 +472,6 @@ def is_lts_driver(device=None):
     return dev.has_aspect_gpu and "1.3" in dev.driver_version
 
 
-def is_dg2(device=None):
-    """
-    Return True if a test is running on DG2 (Intel Arc Alchemist family) GPU device,
-    False otherwise.
-    """
-    return _get_dev_mask(device) in (0x4F00, 0x5600)
-
-
 def is_ptl(device=None):
     """
     Return True if a test is running on Panther Lake with Iris Xe3 GPU device,

dpnp/tests/test_linalg.py

@@ -23,7 +23,7 @@
     get_float_complex_dtypes,
     get_integer_float_dtypes,
     has_support_aspect64,
-    is_dg2,
+    is_arl,
     is_win_platform,
     numpy_version,
 )
@@ -159,7 +159,7 @@ class TestCholesky:
     def test_cholesky(self, array, dtype):
         a = numpy.array(array, dtype=dtype)
         ia = dpnp.array(a)
-        if ia.ndim > 2 and is_win_platform() and is_dg2():
+        if ia.ndim > 2 and is_win_platform() and is_arl():
             pytest.skip("SAT-8206")
         result = dpnp.linalg.cholesky(ia)
         expected = numpy.linalg.cholesky(a)
@@ -180,7 +180,7 @@ def test_cholesky(self, array, dtype):
     @pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))
     def test_cholesky_upper(self, array, dtype):
         ia = dpnp.array(array, dtype=dtype)
-        if ia.ndim > 2 and is_win_platform() and is_dg2():
+        if ia.ndim > 2 and is_win_platform() and is_arl():
             pytest.skip("SAT-8206")
         result = dpnp.linalg.cholesky(ia, upper=True)
 
@@ -224,7 +224,7 @@ def test_cholesky_upper(self, array, dtype):
     def test_cholesky_upper_numpy(self, array, dtype):
         a = numpy.array(array, dtype=dtype)
         ia = dpnp.array(a)
-        if ia.ndim > 2 and is_win_platform() and is_dg2():
+        if ia.ndim > 2 and is_win_platform() and is_arl():
             pytest.skip("SAT-8206")
         result = dpnp.linalg.cholesky(ia, upper=True)
         expected = numpy.linalg.cholesky(a, upper=True)

dpnp/tests/test_sycl_queue.py

@@ -15,7 +15,7 @@
 from .helper import (
     generate_random_numpy_array,
     get_all_dtypes,
-    is_dg2,
+    is_arl,
     is_win_platform,
 )
 
@@ -1493,7 +1493,7 @@ def test_cholesky(self, data, is_empty, device):
         else:
             dtype = dpnp.default_float_type(device)
             x = dpnp.array(data, dtype=dtype, device=device)
-            if x.ndim > 2 and is_win_platform() and is_dg2():
+            if x.ndim > 2 and is_win_platform() and is_arl():
                 pytest.skip("SAT-8206")
 
         result = dpnp.linalg.cholesky(x)

dpnp/tests/test_usm_type.py

@@ -10,7 +10,7 @@
 import dpnp
 from dpnp.dpnp_utils import get_usm_allocations
 
-from .helper import generate_random_numpy_array, is_dg2, is_win_platform
+from .helper import generate_random_numpy_array, is_arl, is_win_platform
 
 list_of_usm_types = ["device", "shared", "host"]
 
@@ -1341,7 +1341,7 @@ def test_cholesky(self, data, is_empty, usm_type):
             x = dpnp.empty(data, dtype=dtype, usm_type=usm_type)
         else:
             x = dpnp.array(data, dtype=dtype, usm_type=usm_type)
-            if x.ndim > 2 and is_win_platform() and is_dg2():
+            if x.ndim > 2 and is_win_platform() and is_arl():
                 pytest.skip("SAT-8206")
 
         result = dpnp.linalg.cholesky(x)

dpnp/tests/third_party/cupy/linalg_tests/test_decomposition.py

@@ -6,8 +6,8 @@
 import dpnp as cupy
 from dpnp.tests.helper import (
     has_support_aspect64,
+    is_arl_or_mtl,
     is_cpu_device,
-    is_dg2,
     is_win_platform,
 )
 from dpnp.tests.third_party.cupy import testing
@@ -84,7 +84,9 @@ def test_decomposition(self, dtype):
         # np.linalg.cholesky only uses a lower triangle of an array
         self.check_L(numpy.array([[1, 2], [1, 9]], dtype))
 
-    @pytest.mark.skipif(is_win_platform() and is_dg2(), reason="SAT-8206")
+    @pytest.mark.skipif(
+        is_win_platform() and is_arl_or_mtl(), reason="SAT-8206"
+    )
     @testing.for_dtypes(
         [
             numpy.int32,