diff options
author | Charles Harris <charlesr.harris@gmail.com> | 2018-06-17 13:38:43 -0600 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-06-17 13:38:43 -0600 |
commit | 621d48d772c9a1c69788cdb68049b729021ba14b (patch) | |
tree | 7cc2e17a5a857cbf3a6c8ad19c9b1828af469d38 | |
parent | 387c08972f2ccc441312006ac68dc4d553d8be8f (diff) | |
parent | 74da42ba633f57306be4f6fedd069ca426254a31 (diff) | |
download | numpy-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.src | 2 |
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]); |