perf(longobject): restore JIT optimizer cases for wide ints

Author: KRRT7

Python/optimizer_bytecodes.c

@@ -396,6 +396,21 @@ dummy_func(void) {
         REPLACE_OPCODE_IF_EVALUATES_PURE(left, right, res);
     }
 
+    op(_BINARY_OP_ADD_INT, (left, right -- res, l, r)) {
+        if (PyJitRef_IsUnique(left)) {
+            REPLACE_OP(this_instr, _BINARY_OP_ADD_INT_INPLACE, 0, 0);
+        }
+        else if (PyJitRef_IsUnique(right)) {
+            REPLACE_OP(this_instr, _BINARY_OP_ADD_INT_INPLACE_RIGHT, 0, 0);
+        }
+        // Result may be a unique compact int or a cached small int
+        // at runtime. Mark as unique; inplace ops verify at runtime.
+        res = PyJitRef_MakeUnique(sym_new_compact_int(ctx));
+        l = left;
+        r = right;
+        REPLACE_OPCODE_IF_EVALUATES_PURE(left, right, res);
+    }
+
     op(_BINARY_OP_ADD_FLOAT, (left, right -- res, l, r)) {
         if (PyJitRef_IsUnique(left)) {
             ADD_OP(_BINARY_OP_ADD_FLOAT_INPLACE, 0, 0);
@@ -432,6 +447,19 @@ dummy_func(void) {
         res = PyJitRef_MakeUnique(sym_new_type(ctx, &PyFloat_Type));
     }
 
+    op(_BINARY_OP_SUBTRACT_INT, (left, right -- res, l, r)) {
+        if (PyJitRef_IsUnique(left)) {
+            REPLACE_OP(this_instr, _BINARY_OP_SUBTRACT_INT_INPLACE, 0, 0);
+        }
+        else if (PyJitRef_IsUnique(right)) {
+            REPLACE_OP(this_instr, _BINARY_OP_SUBTRACT_INT_INPLACE_RIGHT, 0, 0);
+        }
+        res = PyJitRef_MakeUnique(sym_new_compact_int(ctx));
+        l = left;
+        r = right;
+        REPLACE_OPCODE_IF_EVALUATES_PURE(left, right, res);
+    }
+
     op(_BINARY_OP_MULTIPLY_FLOAT, (left, right -- res, l, r)) {
         if (PyJitRef_IsUnique(left)) {
             ADD_OP(_BINARY_OP_MULTIPLY_FLOAT_INPLACE, 0, 0);

Python/optimizer_cases.c.h

@@ -0,0 +1,46 @@
+"""Microbenchmark compact vs wide int add/sub with pyperf.
+
+Use this with PYTHON_JIT=0 and -S if you want a stable interpreter-only run:
+
+    PYTHON_JIT=0 ./python.exe -S Tools/scripts/bench_wide_int_pyperf.py
+"""
+
+from __future__ import annotations
+
+import pyperf
+
+
+def bench_add_compact() -> int:
+    a = 1
+    b = 2
+    return a + b
+
+
+def bench_add_wide() -> int:
+    a = 10_000_000_000
+    b = 1
+    return a + b
+
+
+def bench_sub_compact() -> int:
+    a = 1
+    b = 2
+    return a - b
+
+
+def bench_sub_wide() -> int:
+    a = 10_000_000_000
+    b = 1
+    return a - b
+
+
+def main() -> None:
+    runner = pyperf.Runner()
+    runner.bench_func("add_compact", bench_add_compact)
+    runner.bench_func("add_wide", bench_add_wide)
+    runner.bench_func("sub_compact", bench_sub_compact)
+    runner.bench_func("sub_wide", bench_sub_wide)
+
+
+if __name__ == "__main__":
+    main()