diff options
| -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]); |