Limit the mute only to Iris Xe GPUs

Author: antonwolfy

dpnp/tests/helper.py

@@ -25,6 +25,11 @@ def _assert_shape(a, b):
         # numpy output is scalar, then dpnp is 0-D array
         assert a.shape == (), f"{a.shape} != ()"
 
+def _get_dev_mask(device = None):
+    dev = dpctl.select_default_device() if device is None else device
+    dev_info = dpctl.utils.intel_device_info(dev)
+    return dev_info.get("device_id", 0) & 0xFF00
+
 
 def assert_dtype_allclose(
     dpnp_arr,
@@ -434,6 +439,13 @@ def is_intel_numpy():
     return all(dep["name"].startswith("mkl") for dep in [blas, lapack])
 
 
+def is_iris_xe(device=None):
+    """
+    Return True if a test is running on Iris Xe GPU device, False otherwise.
+    """
+    return _get_dev_mask(device) == 0x9A00
+
+
 def is_lts_driver(device=None):
     """
     Return True if a test is running on a GPU device with LTS driver version,

dpnp/tests/test_arraycreation.py

@@ -19,6 +19,7 @@
     assert_dtype_allclose,
     get_all_dtypes,
     get_array,
+    is_iris_xe,
     is_lts_driver,
     is_win_platform,
 )
@@ -915,7 +916,7 @@ def test_geomspace_num0():
 @pytest.mark.parametrize("num", [2, 4, 8, 3, 9, 27])
 @pytest.mark.parametrize("endpoint", [True, False])
 def test_logspace(dtype, num, endpoint):
-    if not is_win_platform() and is_lts_driver():
+    if not is_win_platform() and is_iris_xe() and is_lts_driver():
         if dpnp.issubdtype(dtype, dpnp.integer) and num in [8, 27] and endpoint is True:
             pytest.skip("SAT-7978")