Skip to content

Commit 65743f0

Browse files
committed
Update internal test f/w to check on strides equality
1 parent 891f135 commit 65743f0

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -171,13 +171,14 @@ def assert_array_equal(
171171
)
172172

173173
if strides_check:
174-
if actual.strides != desired.strides:
174+
strides = tuple(el // desired.itemsize for el in desired.strides)
175+
if actual.strides != strides:
175176
msg = ["Strides are not equal:"]
176177
if err_msg:
177178
msg = [msg[0] + " " + err_msg]
178179
if verbose:
179180
msg.append(" x: {}".format(actual.strides))
180-
msg.append(" y: {}".format(desired.strides))
181+
msg.append(" y: {}".format(strides))
181182
raise AssertionError("\n".join(msg))
182183

183184

0 commit comments

Comments
 (0)