python · Fidget-Spinner · Jun 27, 2025 · Apr 19, 2025 · Apr 19, 2025 · Apr 19, 2025
diff --git a/Include/internal/pycore_opcode_metadata.h b/Include/internal/pycore_opcode_metadata.h
diff --git a/Include/internal/pycore_optimizer.h b/Include/internal/pycore_optimizer.h
@@ -10,6 +10,7 @@ extern "C" {
 
 #include "pycore_typedefs.h"      // _PyInterpreterFrame
 #include "pycore_uop_ids.h"
+#include "pycore_stackref.h"      // _PyStackRef
 #include <stdbool.h>
 
 
@@ -260,6 +261,8 @@ extern bool _Py_uop_sym_is_null(JitOptSymbol *sym);
 extern bool _Py_uop_sym_is_not_null(JitOptSymbol *sym);
 extern bool _Py_uop_sym_is_const(JitOptContext *ctx, JitOptSymbol *sym);
 extern PyObject *_Py_uop_sym_get_const(JitOptContext *ctx, JitOptSymbol *sym);
+extern JitOptSymbol *_Py_uop_sym_new_const_steal(JitOptContext *ctx, PyObject *const_val);
+extern _PyStackRef _Py_uop_sym_get_const_as_stackref(JitOptContext *ctx, JitOptSymbol *sym);
 extern JitOptSymbol *_Py_uop_sym_new_unknown(JitOptContext *ctx);
 extern JitOptSymbol *_Py_uop_sym_new_not_null(JitOptContext *ctx);
 extern JitOptSymbol *_Py_uop_sym_new_type(

diff --git a/Include/internal/pycore_stackref.h b/Include/internal/pycore_stackref.h
@@ -135,6 +135,13 @@ _PyStackRef_FromPyObjectImmortal(PyObject *obj, const char *filename, int linenu
 }
 #define PyStackRef_FromPyObjectImmortal(obj) _PyStackRef_FromPyObjectImmortal(_PyObject_CAST(obj), __FILE__, __LINE__)
 
+static inline _PyStackRef
+_PyStackRef_FromPyObjectImmortalUnchecked(PyObject *obj, const char *filename, int linenumber)
+{
+    return _Py_stackref_create(obj, filename, linenumber);
+}
+#define PyStackRef_FromPyObjectImmortalUnchecked(obj) _PyStackRef_FromPyObjectImmortalUnchecked(_PyObject_CAST(obj), __FILE__, __LINE__)
+
 static inline bool
 PyStackRef_IsTaggedInt(_PyStackRef ref)
 {
@@ -375,6 +382,17 @@ PyStackRef_FromPyObjectImmortal(PyObject *obj)
 }
 #define PyStackRef_FromPyObjectImmortal(obj) PyStackRef_FromPyObjectImmortal(_PyObject_CAST(obj))
 
+static inline _PyStackRef
+PyStackRef_FromPyObjectImmortalUnchecked(PyObject *obj)
+{
+    // Make sure we don't take an already tagged value.
+    assert(((uintptr_t)obj & Py_TAG_BITS) == 0);
+    assert(obj != NULL);
+    return (_PyStackRef){ .bits = (uintptr_t)obj | Py_TAG_DEFERRED };
+}
+#define PyStackRef_FromPyObjectImmortalUnchecked(obj) PyStackRef_FromPyObjectImmortalUnchecked(_PyObject_CAST(obj))
+
+
 #define PyStackRef_CLOSE(REF)                                        \
         do {                                                            \
             _PyStackRef _close_tmp = (REF);                             \
@@ -587,6 +605,12 @@ PyStackRef_FromPyObjectImmortal(PyObject *obj)
     return (_PyStackRef){ .bits = (uintptr_t)obj | Py_TAG_REFCNT};
 }
 
+static inline _PyStackRef
+PyStackRef_FromPyObjectImmortalUnchecked(PyObject *obj)
+{
+    return (_PyStackRef){ .bits = (uintptr_t)obj | Py_TAG_REFCNT};
+}
+
 /* WARNING: This macro evaluates its argument more than once */
 #ifdef _WIN32
 #define PyStackRef_DUP(REF) \

diff --git a/Include/internal/pycore_uop_metadata.h b/Include/internal/pycore_uop_metadata.h
diff --git a/Lib/test/test_generated_cases.py b/Lib/test/test_generated_cases.py
@@ -1972,12 +1972,12 @@ def run_cases_test(self, input: str, input2: str, expected: str):
 
     def test_overridden_abstract(self):
         input = """
-        pure op(OP, (--)) {
+        op(OP, (--)) {
             SPAM();
         }
         """
         input2 = """
-        pure op(OP, (--)) {
+        op(OP, (--)) {
             eggs();
         }
         """
@@ -1991,7 +1991,7 @@ def test_overridden_abstract(self):
 
     def test_overridden_abstract_args(self):
         input = """
-        pure op(OP, (arg1 -- out)) {
+        op(OP, (arg1 -- out)) {
             out = SPAM(arg1);
         }
         op(OP2, (arg1 -- out)) {
@@ -2024,16 +2024,16 @@ def test_overridden_abstract_args(self):
 
     def test_no_overridden_case(self):
         input = """
-        pure op(OP, (arg1 -- out)) {
+        op(OP, (arg1 -- out)) {
             out = SPAM(arg1);
         }
 
-        pure op(OP2, (arg1 -- out)) {
+        op(OP2, (arg1 -- out)) {
         }
 
         """
         input2 = """
-        pure op(OP2, (arg1 -- out)) {
+        op(OP2, (arg1 -- out)) {
             out = NULL;
         }
         """

diff --git a/Misc/NEWS.d/next/Core_and_Builtins/2025-04-19-16-22-47.gh-issue-132732.jgqhlF.rst b/Misc/NEWS.d/next/Core_and_Builtins/2025-04-19-16-22-47.gh-issue-132732.jgqhlF.rst
@@ -0,0 +1 @@
+Automatically constant evaluate bytecode operations marked as pure in the JIT optimizer.
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Automatically constant evaluate bytecode operations marked as pure in the JIT optimizer.
Fidget-Spinner marked this conversation as resolved. Show resolved Hide resolved