Skip to content
Merged
Show file tree
Hide file tree
Changes from all 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 @@ -404,6 +404,14 @@ def has_support_aspect64(device=None):
return dev.has_aspect_fp64


def is_arl_or_mtl(device=None):
"""
Return True if a test is running on Arrow Lake or Meteor Lake GPU device,
False otherwise.
"""
return _get_dev_mask(device) == 0x7D00


def is_cpu_device(device=None):
"""
Return True if a test is running on CPU device, False otherwise.
Expand Down
11 changes: 10 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_arl_or_mtl,
is_win_platform,
numpy_version,
)
from .third_party.cupy import testing
Expand Down Expand Up @@ -109,6 +110,8 @@ def test_usm_ndarray_linalg_batch(func, gen_kwargs, func_kwargs):
)
for _ in range(2)
]
elif func == "cholesky" and is_win_platform() and is_arl_or_mtl():
pytest.skip("SAT-8206")
else:
dpt_args = [
dpt.asarray(generate_random_numpy_array(shape, **gen_kwargs))
Expand Down Expand Up @@ -158,6 +161,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_arl_or_mtl():
pytest.skip("SAT-8206")
result = dpnp.linalg.cholesky(ia)
expected = numpy.linalg.cholesky(a)
assert_dtype_allclose(result, expected)
Expand All @@ -177,6 +182,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_arl_or_mtl():
pytest.skip("SAT-8206")
result = dpnp.linalg.cholesky(ia, upper=True)

if ia.ndim > 2:
Expand Down Expand Up @@ -219,6 +226,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_arl_or_mtl():
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_arl_or_mtl,
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_arl_or_mtl():
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_arl_or_mtl, 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_arl_or_mtl():
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 @@ -6,7 +6,9 @@
import dpnp as cupy
from dpnp.tests.helper import (
has_support_aspect64,
is_arl_or_mtl,
is_cpu_device,
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,9 @@ 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_arl_or_mtl(), reason="SAT-8206"
)
@testing.for_dtypes(
[
numpy.int32,
Expand Down
Loading