Skip to content

Commit d65ad95

Browse files
authored
Enable tests for dpnp.einsum since dpctl issue is resolved (#2425)
[dpctl-2055](IntelPython/dpctl#2055) has been resolved by [dpctl-2058](IntelPython/dpctl#2058). This PR enables previously muted tests for `dpnp.einsum` failing due to dpctl issue.
1 parent eddd826 commit d65ad95

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

dpnp/tests/test_product.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,7 @@ def test_ones(self, dtype):
199199
@pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))
200200
def test_arange(self, dtype):
201201
n = 10**2
202-
m = 10**3 if dtype is not dpnp.float32 else 10**2
202+
m = 10**3 if dtype not in [dpnp.float32, dpnp.complex64] else 10**2
203203
a = numpy.hstack((numpy.arange(n, dtype=dtype),) * m)
204204
b = numpy.flipud(a)
205205
ia = dpnp.array(a)

dpnp/tests/third_party/cupy/linalg_tests/test_einsum.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -373,8 +373,6 @@ def test_einsum_unary_views(self, xp, dtype):
373373
def test_einsum_unary_dtype(self, xp, dtype_a, dtype_out):
374374
if not numpy.can_cast(dtype_a, dtype_out):
375375
pytest.skip()
376-
if cupy.issubdtype(dtype_a, cupy.unsignedinteger):
377-
pytest.skip("dpctl-2055")
378376
a = testing.shaped_arange(self.shape_a, xp, dtype_a)
379377
return xp.einsum(self.subscripts, a, dtype=dtype_out)
380378

0 commit comments

Comments
 (0)