Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions dpnp/tests/helper.py
Original file line number Diff line number Diff line change
Expand Up @@ -464,6 +464,14 @@ 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,
Expand Down
9 changes: 8 additions & 1 deletion dpnp/tests/test_linalg.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,8 @@
get_float_complex_dtypes,
get_integer_float_dtypes,
has_support_aspect64,
is_cpu_device,
is_dg2,
is_win_platform,
numpy_version,
)
from .third_party.cupy import testing
Expand Down Expand Up @@ -158,6 +159,8 @@ 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():
pytest.skip("SAT-8206")
result = dpnp.linalg.cholesky(ia)
expected = numpy.linalg.cholesky(a)
assert_dtype_allclose(result, expected)
Expand All @@ -177,6 +180,8 @@ 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():
pytest.skip("SAT-8206")
result = dpnp.linalg.cholesky(ia, upper=True)

if ia.ndim > 2:
Expand Down Expand Up @@ -219,6 +224,8 @@ 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():
pytest.skip("SAT-8206")
result = dpnp.linalg.cholesky(ia, upper=True)
expected = numpy.linalg.cholesky(a, upper=True)
assert_dtype_allclose(result, expected)
Expand Down
9 changes: 8 additions & 1 deletion dpnp/tests/test_sycl_queue.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,12 @@
from dpnp.dpnp_array import dpnp_array
from dpnp.dpnp_utils import get_usm_allocations

from .helper import generate_random_numpy_array, get_all_dtypes, is_win_platform
from .helper import (
generate_random_numpy_array,
get_all_dtypes,
is_dg2,
is_win_platform,
)

list_of_backend_str = ["cuda", "host", "level_zero", "opencl"]

Expand Down Expand Up @@ -1488,6 +1493,8 @@ 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():
pytest.skip("SAT-8206")

result = dpnp.linalg.cholesky(x)
assert_sycl_queue_equal(result.sycl_queue, x.sycl_queue)
Expand Down
4 changes: 3 additions & 1 deletion dpnp/tests/test_usm_type.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
import dpnp
from dpnp.dpnp_utils import get_usm_allocations

from .helper import generate_random_numpy_array, is_win_platform
from .helper import generate_random_numpy_array, is_dg2, is_win_platform

list_of_usm_types = ["device", "shared", "host"]

Expand Down Expand Up @@ -1341,6 +1341,8 @@ 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():
pytest.skip("SAT-8206")

result = dpnp.linalg.cholesky(x)
assert x.usm_type == result.usm_type
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@
from dpnp.tests.helper import (
has_support_aspect64,
is_cpu_device,
is_dg2,
is_win_platform,
)
from dpnp.tests.third_party.cupy import testing
from dpnp.tests.third_party.cupy.testing import _condition
Expand Down Expand Up @@ -82,6 +84,7 @@ 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")
@testing.for_dtypes(
[
numpy.int32,
Expand Down
Loading