Issue raised as follow-up to: <a class="issue-link js-issue-link" data-error-text="Fai

Probably a duplicate of <a class="issue-link js-issue-link" data-error-text="Failed to

[BUG] Segmentation fault when working on a transposed numpy array with first dimension 1. about bottleneck HOT 2 OPEN

RubendeBruin commented on July 30, 2024

[BUG] Segmentation fault when working on a transposed numpy array with first dimension 1.

from bottleneck.

Comments (2)

cgohlke commented on July 30, 2024

The issue seems that the transposed array is F_CONTIGUOUS and bottleneck wrongly uses the first non-zero stride for iteration. This patch fixes this issue for me ~~but I have not tested it thoroughly~~ on Windows:

diff --git a/bottleneck/include/iterators.h b/bottleneck/include/iterators.h
index 1f5120a..c70c46a 100644
--- a/bottleneck/include/iterators.h
+++ b/bottleneck/include/iterators.h
@@ -108,7 +108,7 @@ static inline void init_iter_all(iter *it, PyArrayObject *a, int ravel, int anyo
         for (i = ndim - 1; i > -1; i--) {
             /* protect against length zero  strides such as in
        * np.ones((2, 2))[..., np.newaxis] */
-            if (strides[i] == 0) {
+            if ((strides[i] == 0) || (shape[i] == 1)) {
                 continue;
             }
             it->astride = strides[i];
@@ -122,7 +122,7 @@ static inline void init_iter_all(iter *it, PyArrayObject *a, int ravel, int anyo
             for (i = 0; i < ndim; i++) {
                 /* protect against length zero  strides such as in
          * np.ones((2, 2), order='F')[np.newaxis, ...] */
-                if (strides[i] == 0) {
+                if ((strides[i] == 0) || (shape[i] == 1)) {
                     continue;
                 }
                 it->astride = strides[i];

The shape, strides, and flags of the transposed array are:

(1, 121, 192)

(185856, 8, 968)

  C_CONTIGUOUS : False
  F_CONTIGUOUS : True
  OWNDATA : False
  WRITEABLE : True
  ALIGNED : True
  WRITEBACKIFCOPY : False
  UPDATEIFCOPY : False

from bottleneck.

cgohlke commented on July 30, 2024

Probably a duplicate of #381 and fixed by #382.

from bottleneck.

Recommend Projects