Skip to content

Commit d31166b

Browse files
committed
Minor fix in cupy tests
1 parent eb71cc5 commit d31166b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/third_party/cupy/testing/helper.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1082,7 +1082,7 @@ def shaped_arange(shape, xp=dpnp, dtype=numpy.float64, order='C'):
10821082
10831083
"""
10841084
dtype = numpy.dtype(dtype)
1085-
a = dpnp.arange(1, prod(shape) + 1, 1)
1085+
a = xp.arange(1, prod(shape) + 1, 1)
10861086
if dtype == '?':
10871087
a = a % 2 == 0
10881088
elif dtype.kind == 'c':
@@ -1108,8 +1108,8 @@ def shaped_reverse_arange(shape, xp=dpnp, dtype=numpy.float32):
11081108
"""
11091109
dtype = numpy.dtype(dtype)
11101110
size = prod(shape)
1111-
# a = dpnp.arange(size, 0, -1)
1112-
a = dpnp.arange(0, size)
1111+
# a = xp.arange(size, 0, -1)
1112+
a = xp.arange(0, size)
11131113
if dtype == '?':
11141114
a = a % 2 == 0
11151115
elif dtype.kind == 'c':

0 commit comments

Comments
 (0)