summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSebastian Berg <sebastian@sipsolutions.net>2015-06-05 19:53:07 +0200
committerSebastian Berg <sebastian@sipsolutions.net>2015-06-05 21:49:02 +0200
commit5bfc31c178faf8470d90ee8594f4de7cb70ee3db (patch)
tree06567c30e7b0477ff3f583495297e850770e07f8
parentc5cf20ab673c76f4b3ef7f85b4d1fd4149237772 (diff)
downloadnumpy-5bfc31c178faf8470d90ee8594f4de7cb70ee3db.tar.gz
BUG: Einsum specialization did not check contiguity of output array
Closes gh-5907
-rw-r--r--numpy/core/src/multiarray/einsum.c.src4
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/core/src/multiarray/einsum.c.src b/numpy/core/src/multiarray/einsum.c.src
index 5b1bb77fa..7483fb01b 100644
--- a/numpy/core/src/multiarray/einsum.c.src
+++ b/numpy/core/src/multiarray/einsum.c.src
@@ -1770,14 +1770,14 @@ get_sum_of_products_function(int nop, int type_num,
}
/* Check for all contiguous */
- for (iop = 0; iop < nop; ++iop) {
+ for (iop = 0; iop < nop + 1; ++iop) {
if (fixed_strides[iop] != itemsize) {
break;
}
}
/* Contiguous loop */
- if (iop == nop) {
+ if (iop == nop + 1) {
return _allcontig_specialized_table[type_num][nop <= 3 ? nop : 0];
}