bpo-34303: Micro-optimizations in functools.reduce() (GH-8598)

diff --git a/Misc/NEWS.d/next/Library/2018-08-03-09-47-20.bpo-34303.tOE2HP.rst b/Misc/NEWS.d/next/Library/2018-08-03-09-47-20.bpo-34303.tOE2HP.rst
new file mode 100644 (file)
index 0000000..94c1299
--- /dev/null
+++ b/Misc/NEWS.d/next/Library/2018-08-03-09-47-20.bpo-34303.tOE2HP.rst
@@ -0,0 +1,2 @@
+Performance of :func:`functools.reduce` is slightly improved. Patch by
+Sergey Fedoseev.

diff --git a/Modules/_functoolsmodule.c b/Modules/_functoolsmodule.c
index aca5bad23f585ba8ee1c8e3e654eb0077f0f267f..a101363bf02a07dedfa3184a52fc42c46bbb660c 100644 (file)
--- a/Modules/_functoolsmodule.c
+++ b/Modules/_functoolsmodule.c
@@ -626,10 +626,13 @@ functools_reduce(PyObject *self, PyObject *args)
         if (result == NULL)
             result = op2;
         else {
-            PyTuple_SetItem(args, 0, result);
-            PyTuple_SetItem(args, 1, op2);
-            if ((result = PyEval_CallObject(func, args)) == NULL)
+            /* Update the args tuple in-place */
+            assert(args->ob_refcnt == 1);
+            Py_XSETREF(_PyTuple_ITEMS(args)[0], result);
+            Py_XSETREF(_PyTuple_ITEMS(args)[1], op2);
+            if ((result = PyObject_Call(func, args, NULL)) == NULL) {
                 goto Fail;
+            }
         }
     }