Follow up gh-2728 and extend muting to any GPU with non LTS2 version

Author: antonwolfy

dpnp/tests/helper.py

@@ -1,4 +1,5 @@
 import importlib.util
+from enum import Enum
 from sys import platform
 
 import dpctl
@@ -11,6 +12,11 @@
 from . import config
 
 
+class LTS_VERSION(Enum):
+    V1_3 = "1.3"
+    V1_6 = "1.6"
+
+
 def _assert_dtype(a_dt, b_dt, check_only_type_kind=False):
     if check_only_type_kind:
         assert a_dt.kind == b_dt.kind, f"{a_dt.kind} != {b_dt.kind}"
@@ -475,13 +481,13 @@ def is_lnl(device=None):
     return _get_dev_mask(device) == 0x6400
 
 
-def is_lts_driver(device=None):
+def is_lts_driver(version=LTS_VERSION.V1_3, device=None):
     """
     Return True if a test is running on a GPU device with LTS driver version,
     False otherwise.
     """
     dev = dpctl.select_default_device() if device is None else device
-    return dev.has_aspect_gpu and "1.3" in dev.driver_version
+    return dev.has_aspect_gpu and version.value in dev.driver_version
 
 
 def is_ptl(device=None):

dpnp/tests/test_mathematical.py

@@ -20,6 +20,7 @@
 from dpnp.dpnp_utils import map_dtype_to_device
 
 from .helper import (
+    LTS_VERSION,
     assert_dtype_allclose,
     generate_random_numpy_array,
     get_abs_array,
@@ -33,7 +34,7 @@
     has_support_aspect16,
     has_support_aspect64,
     is_intel_numpy,
-    is_ptl,
+    is_lts_driver,
     numpy_version,
 )
 from .third_party.cupy import testing
@@ -218,7 +219,7 @@ def _get_exp_array(self, a, axis, dtype):
     @pytest.mark.parametrize("axis", [None, 2, -1])
     @pytest.mark.parametrize("include_initial", [True, False])
     def test_basic(self, dtype, axis, include_initial):
-        if axis is None and is_ptl():
+        if axis is None and not is_lts_driver(version=LTS_VERSION.V1_6):
             pytest.skip("due to SAT-8336")
 
         a = dpnp.ones((3, 4, 5, 6, 7), dtype=dtype)
@@ -238,7 +239,7 @@ def test_basic(self, dtype, axis, include_initial):
     @pytest.mark.parametrize("axis", [None, 2, -1])
     @pytest.mark.parametrize("include_initial", [True, False])
     def test_include_initial(self, dtype, axis, include_initial):
-        if axis is None and is_ptl():
+        if axis is None and not is_lts_driver(version=LTS_VERSION.V1_6):
             pytest.skip("due to SAT-8336")
 
         a = dpnp.ones((3, 4, 5, 6, 7), dtype=dtype)