From b90ecea9e6b33dae360ed7eb2c32598f98444c4d Mon Sep 17 00:00:00 2001 From: Mark Shannon Date: Thu, 5 Jun 2025 18:53:57 +0100 Subject: GH-132554: Fix tier2 `FOR_ITER` implementation and optimizations (GH-135137) --- Python/optimizer_bytecodes.c | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'Python/optimizer_bytecodes.c') diff --git a/Python/optimizer_bytecodes.c b/Python/optimizer_bytecodes.c index e1209209660..b4220e2c627 100644 --- a/Python/optimizer_bytecodes.c +++ b/Python/optimizer_bytecodes.c @@ -840,6 +840,17 @@ dummy_func(void) { value = sym_new_unknown(ctx); } + op(_GET_ITER, (iterable -- iter, index_or_null)) { + if (sym_matches_type(iterable, &PyTuple_Type) || sym_matches_type(iterable, &PyList_Type)) { + iter = iterable; + index_or_null = sym_new_not_null(ctx); + } + else { + iter = sym_new_not_null(ctx); + index_or_null = sym_new_unknown(ctx); + } + } + op(_FOR_ITER_GEN_FRAME, (unused, unused -- unused, unused, gen_frame: _Py_UOpsAbstractFrame*)) { gen_frame = NULL; /* We are about to hit the end of the trace */ -- cgit v1.2.3