From 78731c01167604cbd60dcfd895669760fa8f5609 Mon Sep 17 00:00:00 2001
From: Serhiy Storchaka <storchaka@gmail.com>
Date: Mon, 16 Jun 2025 11:46:37 +0300
Subject: [PATCH] gh-102221: Optimize math.lcm() for multiple arguments

---
 Modules/mathmodule.c | 57 ++++++++++++++++++++++++--------------------
 1 file changed, 31 insertions(+), 26 deletions(-)

diff --git a/Modules/mathmodule.c b/Modules/mathmodule.c
index 7c2a421dd6a450..5bab94d1ed84bb 100644
--- a/Modules/mathmodule.c
+++ b/Modules/mathmodule.c
@@ -802,41 +802,46 @@ math_lcm_impl(PyObject *module, PyObject * const *args,
               Py_ssize_t args_length)
 /*[clinic end generated code: output=c8a59a5c2e55c816 input=3e4f4b7cdf948a98]*/
 {
-    PyObject *res, *x;
-    Py_ssize_t i;
-
     if (args_length == 0) {
         return PyLong_FromLong(1);
     }
-    res = PyNumber_Index(args[0]);
-    if (res == NULL) {
-        return NULL;
+    PyObject *res;
+    PyObject *stack[8 * sizeof(Py_ssize_t)];
+    int top = 0;
+    Py_ssize_t i = 0;
+    while (1) {
+        size_t j = i;
+        res = PyNumber_Index(args[i++]);
+        if (res == NULL) {
+            goto error;
+        }
+        if (i >= args_length) {
+            j = ((size_t)1 << top) - 1;
+        }
+        for (; j & 1; j >>= 1) {
+            top--;
+            Py_SETREF(res, long_lcm(res, stack[top]));
+            if (res == NULL) {
+                goto error;
+            }
+            Py_DECREF(stack[top]);
+        }
+        if (i >= args_length) {
+            break;
+        }
+        stack[top++] = res;
     }
     if (args_length == 1) {
         Py_SETREF(res, PyNumber_Absolute(res));
-        return res;
     }
+    return res;
 
-    PyObject *zero = _PyLong_GetZero();  // borrowed ref
-    for (i = 1; i < args_length; i++) {
-        x = PyNumber_Index(args[i]);
-        if (x == NULL) {
-            Py_DECREF(res);
-            return NULL;
-        }
-        if (res == zero) {
-            /* Fast path: just check arguments.
-               It is okay to use identity comparison here. */
-            Py_DECREF(x);
-            continue;
-        }
-        Py_SETREF(res, long_lcm(res, x));
-        Py_DECREF(x);
-        if (res == NULL) {
-            return NULL;
-        }
+error:
+    while (top > 0) {
+        top--;
+        Py_DECREF(stack[top]);
     }
-    return res;
+    return NULL;
 }