diff options
Diffstat (limited to 'Objects/frameobject.c')
| -rw-r--r-- | Objects/frameobject.c | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/Objects/frameobject.c b/Objects/frameobject.c index a448ba3..545c1b6 100644 --- a/Objects/frameobject.c +++ b/Objects/frameobject.c @@ -458,22 +458,30 @@ _PyFrame_GetState(PyFrameObject *frame) static void add_load_fast_null_checks(PyCodeObject *co) { + int changed = 0; _Py_CODEUNIT *instructions = _PyCode_CODE(co); for (Py_ssize_t i = 0; i < Py_SIZE(co); i++) { switch (_Py_OPCODE(instructions[i])) { case LOAD_FAST: case LOAD_FAST__LOAD_FAST: case LOAD_FAST__LOAD_CONST: + changed = 1; _Py_SET_OPCODE(instructions[i], LOAD_FAST_CHECK); break; case LOAD_CONST__LOAD_FAST: + changed = 1; _Py_SET_OPCODE(instructions[i], LOAD_CONST); break; case STORE_FAST__LOAD_FAST: + changed = 1; _Py_SET_OPCODE(instructions[i], STORE_FAST); break; } } + if (changed) { + // invalidate cached co_code object + Py_CLEAR(co->_co_code); + } } /* Setter for f_lineno - you can set f_lineno from within a trace function in |