Skip to content

Commit 4811649

Browse files
committed
Use a smaller array
1 parent 96357a0 commit 4811649

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

cuda_core/tests/test_memory.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -616,12 +616,12 @@ def test_strided_memory_view_leak():
616616

617617
def test_strided_memory_view_refcnt():
618618
# Use Fortran ordering so strides is used
619-
a = np.zeros((11171, 4), dtype=np.uint8, order="F")
619+
a = np.zeros((64, 4), dtype=np.uint8, order="F")
620620
av = StridedMemoryView(a, stream_ptr=-1)
621621
# segfaults if refcnt is wrong
622-
assert av.shape[0] == 11171
622+
assert av.shape[0] == 64
623623
assert sys.getrefcount(av.shape) >= 2
624624

625625
assert av.strides[0] == 1
626-
assert av.strides[1] == 11171
626+
assert av.strides[1] == 64
627627
assert sys.getrefcount(av.strides) >= 2

0 commit comments

Comments
 (0)