gh-131798: Add _CHECK_IS_NOT_PY_CALLABLE to the JIT optimizer (GH-148434)

This commit is contained in:
Wulian233 2026-04-12 21:07:19 +08:00 committed by GitHub
parent 30c698a655
commit 03d2f035d6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 28 additions and 0 deletions

View file

@ -2755,6 +2755,21 @@ def testfunc(n):
self.assertNotIn("_GUARD_TOS_INT", uops)
self.assertIn("_POP_TOP_NOP", uops)
def test_check_is_not_py_callable(self):
def testfunc(n):
total = 0
f = len
xs = (1, 2, 3)
for _ in range(n):
total += f(xs)
return total
res, ex = self._run_with_optimizer(testfunc, TIER2_THRESHOLD)
self.assertEqual(res, 3 * TIER2_THRESHOLD)
self.assertIsNotNone(ex)
uops = get_opnames(ex)
self.assertNotIn("_CHECK_IS_NOT_PY_CALLABLE", uops)
def test_call_len_string(self):
def testfunc(n):
for _ in range(n):

View file

@ -1217,6 +1217,13 @@ dummy_func(void) {
(void)framesize;
}
op(_CHECK_IS_NOT_PY_CALLABLE, (callable, unused, unused[oparg] -- callable, unused, unused[oparg])) {
PyTypeObject *type = sym_get_type(callable);
if (type && type != &PyFunction_Type && type != &PyMethod_Type) {
ADD_OP(_NOP, 0, 0);
}
}
op(_PUSH_FRAME, (new_frame -- )) {
SYNC_SP();
if (!CURRENT_FRAME_IS_INIT_SHIM()) {

View file

@ -3651,6 +3651,12 @@
}
case _CHECK_IS_NOT_PY_CALLABLE: {
JitOptRef callable;
callable = stack_pointer[-2 - oparg];
PyTypeObject *type = sym_get_type(callable);
if (type && type != &PyFunction_Type && type != &PyMethod_Type) {
ADD_OP(_NOP, 0, 0);
}
break;
}