diff options
| author | mattip <matti.picus@gmail.com> | 2018-06-17 09:42:09 -0700 |
|---|---|---|
| committer | mattip <matti.picus@gmail.com> | 2018-06-17 09:42:09 -0700 |
| commit | 74da42ba633f57306be4f6fedd069ca426254a31 (patch) | |
| tree | 7cc2e17a5a857cbf3a6c8ad19c9b1828af469d38 | |
| parent | 387c08972f2ccc441312006ac68dc4d553d8be8f (diff) | |
| download | numpy-74da42ba633f57306be4f6fedd069ca426254a31.tar.gz | |
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]); |