Commit 802508cb authored by Mark Florisson's avatar Mark Florisson

Fix strides array bug

parent 9ab6e447
...@@ -336,7 +336,7 @@ cdef class array: ...@@ -336,7 +336,7 @@ cdef class array:
assert idx == self.ndim assert idx == self.ndim
if mode == "fortran": if mode == "fortran":
idx = 0; stride = 1 idx = 0; stride = itemsize
for dim in shape: for dim in shape:
self.strides[idx] = stride self.strides[idx] = stride
int_dim = <Py_ssize_t>dim int_dim = <Py_ssize_t>dim
...@@ -345,7 +345,7 @@ cdef class array: ...@@ -345,7 +345,7 @@ cdef class array:
assert idx == self.ndim assert idx == self.ndim
self.len = stride * self.itemsize self.len = stride * self.itemsize
elif mode == "c": elif mode == "c":
idx = self.ndim-1; stride = 1 idx = self.ndim-1; stride = itemsize
for dim in reversed(shape): for dim in reversed(shape):
self.strides[idx] = stride self.strides[idx] = stride
int_dim = <Py_ssize_t>dim int_dim = <Py_ssize_t>dim
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment