specialize: make specialization thread-safe

colesbury · Apr 23, 2023 · 7e75686 · 7e75686
1 parent 149ea9d
commit 7e75686
Show file tree

Hide file tree

Showing 15 changed files with 549 additions and 223 deletions.
diff --git a/Include/cpython/code.h b/Include/cpython/code.h
@@ -30,7 +30,9 @@ typedef union {
 static inline void
 _py_set_opcode(_Py_CODEUNIT *word, uint8_t opcode)
 {
-    word->opcode = opcode;
+    _Py_CODEUNIT new_word = *word;
+    new_word.opcode = opcode;
+    _Py_atomic_store_uint16(&word->cache, new_word.cache);
 }
 
 #define _Py_SET_OPCODE(word, opcode) _py_set_opocde(&(word), opcode)

diff --git a/Include/internal/pycore_code.h b/Include/internal/pycore_code.h
@@ -99,6 +99,7 @@ struct callable_cache {
     PyObject *len;
     PyObject *list_append;
     PyObject *object__getattribute__;
+    PyObject *object__class__;
 };
 
 /* "Locals plus" for a code object is the set of locals + cell vars +

diff --git a/Include/internal/pycore_opcode.h b/Include/internal/pycore_opcode.h