Skip to content

Commit 2547aed

Browse files
committed
numpy.ndarray mask is not supported in indexing
1 parent 223141d commit 2547aed

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dpnp/tests/third_party/cupy/testing/_loops.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -408,7 +408,9 @@ def test_func(*args, **kw):
408408
if cupy_r.shape == ():
409409
skip = (mask == 0).all()
410410
else:
411-
cupy_r = cupy_r[mask].asnumpy()
411+
# mask is numpy.ndarray here which is not supported now
412+
# TODO remove asarray() once dpctl-2053 is addressed
413+
cupy_r = cupy_r[cupy.asarray(mask)].asnumpy()
412414
numpy_r = numpy_r[mask]
413415

414416
if not skip:

0 commit comments

Comments
 (0)