deepfreeze use preallocated small ints

python · gvanrossum · Jan 20, 2022 · Jan 20, 2022 · Jan 20, 2022 · Jan 20, 2022
commit 7fb3f2295534a3cc0ce45779fc8f070e506e554d
@@ -113,6 +113,7 @@ def __init__(self, file: TextIO):
         self.write('#include "Python.h"')
         self.write('#include "internal/pycore_gc.h"')
         self.write('#include "internal/pycore_code.h"')
+        self.write('#include "internal/pycore_long.h"')
         self.write("")
 
     @contextlib.contextmanager
@@ -313,6 +314,8 @@ def _generate_int_for_bits(self, name: str, i: int, digit: int) -> None:
                 self.write(f".ob_digit = {{ {ds} }},")
 
     def generate_int(self, name: str, i: int) -> str:
+        if -5 <= i <= 256:
+            return f"(PyObject *)&_PyLong_SMALL_INTS[_PY_NSMALLNEGINTS + {i}]"
         if abs(i) < 2**15:
             self._generate_int_for_bits(name, i, 2**15)
         else: