BUG: fix a leak in ufunc_object.PyUFunc_RegisterLoopForDescr

Hugo Defois · Hugo Defois · commit 4b971415cd44 · 2021-07-05T16:37:11.000+02:00
diff --git a/numpy/core/src/umath/ufunc_object.c b/numpy/core/src/umath/ufunc_object.c
@@ -5218,6 +5218,7 @@ PyUFunc_RegisterLoopForDescr(PyUFuncObject *ufunc,
 
     arg_typenums = PyArray_malloc(ufunc->nargs * sizeof(int));
     if (arg_typenums == NULL) {
+        Py_DECREF(key);
         PyErr_NoMemory();
         return -1;
     }

Original file line number	Diff line number	Diff line change
`@@ -5218,6 +5218,7 @@ PyUFunc_RegisterLoopForDescr(PyUFuncObject *ufunc,`
`5218`	`5218`
`5219`	`5219`	`arg_typenums = PyArray_malloc(ufunc->nargs * sizeof(int));`
`5220`	`5220`	`if (arg_typenums == NULL) {`
	`5221`	`+ Py_DECREF(key);`
`5221`	`5222`	`PyErr_NoMemory();`
`5222`	`5223`	`return -1;`
`5223`	`5224`	`}`