add free

Author: iritkatriel

Include/internal/pycore_code.h

@@ -482,13 +482,18 @@ adaptive_counter_backoff(_Py_BackoffCounter counter) {
 /* Specialization Extensions */
 
 /* callbacks for an external specialization */
+
+struct _PyBinopSpecializationDescr;
+
 typedef int (*binaryopguardfunc)(PyObject *lhs, PyObject *rhs);
-typedef PyObject *(*binaryopactionfunc)(PyObject *lhs, PyObject *rhs);
+typedef PyObject* (*binaryopactionfunc)(PyObject *lhs, PyObject *rhs);
+typedef void (*binaryopfreefunc)(struct _PyBinopSpecializationDescr *descr);
 
 typedef struct _PyBinopSpecializationDescr {
     int oparg;
     binaryopguardfunc guard;
     binaryopactionfunc action;
+    binaryopfreefunc free;
 } _PyBinaryOpSpecializationDescr;
 
 /* Comparison bit masks. */

Modules/arraymodule.c

@@ -3024,6 +3024,14 @@ array_subscr_action(PyObject *lhs, PyObject *rhs)
     return array_subscr(lhs, rhs);
 }
 
+static void
+array_subscr_free(_PyBinaryOpSpecializationDescr* descr)
+{
+    if (descr != NULL) {
+        PyMem_Free(descr);
+    }
+}
+
 static int
 array_binop_specialize(PyObject *v, PyObject *w, int oparg,
                        _PyBinaryOpSpecializationDescr **descr)
@@ -3048,6 +3056,7 @@ array_binop_specialize(PyObject *v, PyObject *w, int oparg,
                     .oparg = oparg,
                     .guard = array_subscr_guard,
                     .action = array_subscr_action,
+                    .free = array_subscr_free,
                 };
                 return 1;
             }

Python/bytecodes.c

@@ -805,7 +805,12 @@ dummy_func(
             assert(d->guard);
             int res = d->guard(left_o, right_o);
             ERROR_IF(res < 0);
-            DEOPT_IF(res == 0);
+            if (res == 0) {
+                if (d->free) {
+                    d->free(d);
+                }
+                DEOPT_IF(true);
+            }
         }
 
         pure op(_BINARY_OP_EXTEND, (descr/4, left, right -- res)) {