Skip to content

Commit 848dfcc

Browse files
committed
refactor: count backwards instead of reversing
1 parent 1729d82 commit 848dfcc

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

numba_cuda/numba/cuda/api_util.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,9 @@ def _fill_stride_by_order(shape, dtype, order):
3636
strides = [0] * ndims
3737
if order == "C":
3838
strides[-1] = dtype.itemsize
39-
for d in reversed(range(ndims - 1)):
39+
# -2 because we subtract one for zero-based indexing and another one
40+
# for skipping the already-filled-in last element
41+
for d in range(ndims - 2, -1, -1):
4042
strides[d] = strides[d + 1] * shape[d + 1]
4143
elif order == "F":
4244
strides[0] = dtype.itemsize

0 commit comments

Comments
 (0)