diff --git a/dpnp/tests/test_mathematical.py b/dpnp/tests/test_mathematical.py index 77c65991e9c..1418a5a7fed 100644 --- a/dpnp/tests/test_mathematical.py +++ b/dpnp/tests/test_mathematical.py @@ -33,6 +33,7 @@ has_support_aspect16, has_support_aspect64, is_intel_numpy, + is_ptl, numpy_version, ) from .third_party.cupy import testing @@ -217,6 +218,9 @@ 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(): + pytest.skip("due to SAT-8336") + a = dpnp.ones((3, 4, 5, 6, 7), dtype=dtype) res = dpnp.cumlogsumexp(a, axis=axis, include_initial=include_initial) @@ -234,6 +238,9 @@ 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(): + pytest.skip("due to SAT-8336") + a = dpnp.ones((3, 4, 5, 6, 7), dtype=dtype) if dpnp.issubdtype(a, dpnp.float32):