summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorCharles Harris <charlesr.harris@gmail.com>2018-06-17 13:38:43 -0600
committerGitHub <noreply@github.com>2018-06-17 13:38:43 -0600
commit621d48d772c9a1c69788cdb68049b729021ba14b (patch)
tree7cc2e17a5a857cbf3a6c8ad19c9b1828af469d38
parent387c08972f2ccc441312006ac68dc4d553d8be8f (diff)
parent74da42ba633f57306be4f6fedd069ca426254a31 (diff)
downloadnumpy-621d48d772c9a1c69788cdb68049b729021ba14b.tar.gz
Merge pull request #11365 from mattip/nditer-einsum
BUG: add missing NpyIter_Close in einsum
-rw-r--r--numpy/core/src/multiarray/einsum.c.src2
1 files changed, 2 insertions, 0 deletions
diff --git a/numpy/core/src/multiarray/einsum.c.src b/numpy/core/src/multiarray/einsum.c.src
index 33184d99a..039bcc15e 100644
--- a/numpy/core/src/multiarray/einsum.c.src
+++ b/numpy/core/src/multiarray/einsum.c.src
@@ -2857,6 +2857,7 @@ PyArray_EinsteinSum(char *subscripts, npy_intp nop,
iternext = NpyIter_GetIterNext(iter, NULL);
if (iternext == NULL) {
+ NpyIter_Close(iter);
NpyIter_Deallocate(iter);
Py_DECREF(ret);
goto fail;
@@ -2880,6 +2881,7 @@ PyArray_EinsteinSum(char *subscripts, npy_intp nop,
}
finish:
+ NpyIter_Close(iter);
NpyIter_Deallocate(iter);
for (iop = 0; iop < nop; ++iop) {
Py_DECREF(op[iop]);