Address code review

Author: corona10

Include/internal/pycore_uop.h

@@ -37,6 +37,7 @@ typedef struct _PyUOpInstruction{
 
 // This is the length of the trace we project initially.
 #define UOP_MAX_TRACE_LENGTH 1200
+#define UOP_BUFFER_SIZE (UOP_MAX_TRACE_LENGTH * sizeof(_PyUOpInstruction))
 
 #ifdef __cplusplus
 }

Python/optimizer.c

@@ -1282,7 +1282,7 @@ uop_optimize(
     _Py_BloomFilter_Init(&dependencies);
     PyInterpreterState *interp = _PyInterpreterState_GET();
     if (interp->jit_uop_buffer == NULL) {
-        interp->jit_uop_buffer = (_PyUOpInstruction *)PyMem_RawMalloc(UOP_MAX_TRACE_LENGTH*sizeof(_PyUOpInstruction));
+        interp->jit_uop_buffer = (_PyUOpInstruction *)_PyObject_VirtualAlloc(UOP_BUFFER_SIZE);
         if (interp->jit_uop_buffer == NULL) {
             return 0;
         }

Python/pystate.c

@@ -23,6 +23,7 @@
 #include "pycore_runtime_init.h"  // _PyRuntimeState_INIT
 #include "pycore_stackref.h"      // Py_STACKREF_DEBUG
 #include "pycore_time.h"          // _PyTime_Init()
+#include "pycore_uop.h".          // UOP_BUFFER_SIZE
 #include "pycore_uniqueid.h"      // _PyObject_FinalizePerThreadRefcounts()
 
 
@@ -809,7 +810,7 @@ interpreter_clear(PyInterpreterState *interp, PyThreadState *tstate)
 #ifdef _Py_TIER2
     _Py_ClearExecutorDeletionList(interp);
     if (interp->jit_uop_buffer != NULL) {
-        PyMem_RawFree(interp->jit_uop_buffer);
+        _PyObject_VirtualFree(interp->jit_uop_buffer, UOP_BUFFER_SIZE);
         interp->jit_uop_buffer = NULL;
     }
 #endif