diff --git a/dpnp/tests/helper.py b/dpnp/tests/helper.py index 65917ee6c34..c0778777ec5 100644 --- a/dpnp/tests/helper.py +++ b/dpnp/tests/helper.py @@ -430,6 +430,14 @@ def is_arl_or_mtl(device=None): return _get_dev_mask(device) == 0x7D00 +def is_bmg(device=None): + """ + Return True if a test is running on Arc Battlemage (B-Series) GPU device, + False otherwise. + """ + return _get_dev_mask(device) == 0xE200 + + def is_cpu_device(device=None): """ Return True if a test is running on CPU device, False otherwise. diff --git a/dpnp/tests/test_histogram.py b/dpnp/tests/test_histogram.py index 6920e2d8803..b0988609d2c 100644 --- a/dpnp/tests/test_histogram.py +++ b/dpnp/tests/test_histogram.py @@ -11,6 +11,7 @@ import dpnp from .helper import ( + LTS_VERSION, assert_dtype_allclose, generate_random_numpy_array, get_abs_array, @@ -21,6 +22,10 @@ get_integer_dtypes, get_integer_float_dtypes, has_support_aspect64, + is_bmg, + is_lnl, + is_lts_driver, + is_win_platform, numpy_version, ) from .third_party.cupy import testing @@ -487,6 +492,12 @@ def test_weights_another_sycl_queue(self): with assert_raises(ValueError): dpnp.histogram(v, weights=w) + @pytest.mark.skipif( + not is_win_platform() + and not is_lts_driver(version=LTS_VERSION.V1_6) + and (is_lnl() or is_bmg()), + reason="SAT-8135", + ) @pytest.mark.parametrize( "bins_count", [10, 10**2, 10**3, 10**4, 10**5, 10**6], @@ -585,6 +596,12 @@ def test_weights_unsupported_dtype(self, xp, dt): w = xp.arange(5, dtype=dt) assert_raises((TypeError, ValueError), xp.bincount, v, weights=w) + @pytest.mark.skipif( + not is_win_platform() + and not is_lts_driver(version=LTS_VERSION.V1_6) + and (is_lnl() or is_bmg()), + reason="SAT-8135", + ) @pytest.mark.parametrize( "bins_count", [10, 10**2, 10**3, 10**4, 10**5, 10**6], @@ -851,6 +868,12 @@ def test_weights_another_sycl_queue(self): with assert_raises(ValueError): dpnp.histogramdd(v, weights=w) + @pytest.mark.skipif( + not is_win_platform() + and not is_lts_driver(version=LTS_VERSION.V1_6) + and (is_lnl() or is_bmg()), + reason="SAT-8135", + ) @pytest.mark.parametrize( "bins_count", [10, 10**2, 10**3, 10**4, 10**5, 10**6], @@ -1092,6 +1115,12 @@ def test_size_mismatch(self, xp): y = xp.linspace(0.0, 1.0, num=20) assert_raises(ValueError, xp.histogram2d, x, y) + @pytest.mark.skipif( + not is_win_platform() + and not is_lts_driver(version=LTS_VERSION.V1_6) + and (is_lnl() or is_bmg()), + reason="SAT-8135", + ) @pytest.mark.parametrize( "bins_count", [10, 10**2, 10**3],