Skip to content

Commit 46470ca

Browse files
authored
Fix NVIDIA#1417: Fix test for Numpy 2.4.0 (NVIDIA#1418)
* Fix NVIDIA#1417: Fix test for Numpy 2.4.0 * Slightly improve logic
1 parent 9e9b4c8 commit 46470ca

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

cuda_core/tests/test_utils.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -95,10 +95,7 @@ def _check_view(self, view, in_arr):
9595
assert view.shape == in_arr.shape
9696
assert view.size == in_arr.size
9797
strides_in_counts = convert_strides_to_counts(in_arr.strides, in_arr.dtype.itemsize)
98-
if in_arr.flags.c_contiguous:
99-
assert view.strides is None
100-
else:
101-
assert view.strides == strides_in_counts
98+
assert (in_arr.flags.c_contiguous and view.strides is None) or view.strides == strides_in_counts
10299
assert view.dtype == in_arr.dtype
103100
assert view.device_id == -1
104101
assert view.is_device_accessible is False

0 commit comments

Comments
 (0)