Skip to content

Commit 6ba840a

Browse files
Update test_solve in test_sycl_queue (#2126)
Co-authored-by: Anton <100830759+antonwolfy@users.noreply.github.com>
1 parent e8f348b commit 6ba840a

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/test_sycl_queue.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2251,8 +2251,13 @@ def test_solve(matrix, vector, device):
22512251
a_dp = dpnp.array(a_np, device=device)
22522252
b_dp = dpnp.array(b_np, device=device)
22532253

2254-
if a_dp.ndim > 2 and a_dp.device.sycl_device.is_cpu:
2255-
pytest.skip("SAT-6842: reported hanging in public CI")
2254+
# In numpy 2.0 the broadcast ambiguity has been removed and now
2255+
# b is treaded as a single vector if and only if it is 1-dimensional;
2256+
# for other cases this signature must be followed
2257+
# (..., m, m), (..., m, n) -> (..., m, n)
2258+
# https://github.com/numpy/numpy/pull/25914
2259+
if a_dp.ndim > 2 and numpy.lib.NumpyVersion(numpy.__version__) >= "2.0.0":
2260+
pytest.skip("SAT-6928")
22562261

22572262
result = dpnp.linalg.solve(a_dp, b_dp)
22582263
expected = numpy.linalg.solve(a_np, b_np)

0 commit comments

Comments
 (0)