re-enable jit on some stuff

Author: Fidget-Spinner

Lib/test/test_capi/test_opt.py

@@ -417,8 +417,6 @@ def testfunc(n, m):
         self.assertIn("_FOR_ITER_TIER_TWO", uops)
 
 
-@requires_specialization
-@unittest.skipIf(Py_GIL_DISABLED, "optimizer not yet supported in free-threaded builds")
 @requires_jit_enabled
 @unittest.skipIf(os.getenv("PYTHON_UOPS_OPTIMIZE") == "0", "Needs uop optimizer to run.")
 class TestUopsOptimization(unittest.TestCase):

Python/pystate.c

@@ -1844,6 +1844,13 @@ tstate_delete_common(PyThreadState *tstate, int release_gil)
         _PyObject_VirtualFree(_tstate->jit_tracer_state.code_buffer, UOP_BUFFER_SIZE);
         _tstate->jit_tracer_state.code_buffer = NULL;
     }
+#ifdef Py_GIL_DISABLED
+    // There's only one thread. Re-enable JIT.
+    PyThreadState *curr = interp->threads.head;
+    if (curr != NULL && curr->prev == NULL && curr->next == NULL) {
+        _Py_atomic_store_char_relaxed(&((_PyThreadStateImpl*)curr)->jit_executor_state.jit, true);
+    }
+#endif
 #endif
 
     HEAD_UNLOCK(runtime);