gh-131798: Add _GET_ITER_TRAD to JIT optimizer

Author: Wulian233

Lib/test/test_capi/test_opt.py

@@ -438,7 +438,25 @@ def testfunc(n):
         self.assertNotIn("_GET_ITER", uops)
         self.assertNotIn("_GET_ITER_VIRTUAL", uops)
         self.assertNotIn("_GET_ITER_SELF", uops)
+        self.assertNotIn("_GUARD_NOS_NULL", uops)
+
+    def test_get_iter_trad_for_iter_tier_two(self):
+        d = {v: v for v in range(20)}
 
+        def testfunc(n):
+            total = 0
+            while n:
+                n -= 1
+                for _ in d:
+                    total += 1
+            return total
+
+        self.assertEqual(testfunc(TIER2_THRESHOLD), TIER2_THRESHOLD * len(d))
+        ex = get_first_executor(testfunc)
+        self.assertIsNotNone(ex)
+        uops = get_opnames(ex)
+        self.assertIn("_FOR_ITER_TIER_TWO", uops)
+        self.assertNotIn("_GET_ITER_TRAD", uops)
 
     def test_for_iter_range(self):
         def testfunc(n):

Python/optimizer_bytecodes.c

@@ -1443,6 +1443,12 @@ dummy_func(void) {
         }
     }
 
+    op(_GET_ITER_TRAD, (iterable -- iter, index_or_null)) {
+        (void)iterable;
+        iter = sym_new_not_null(ctx);
+        index_or_null = sym_new_null(ctx);
+    }
+
     op(_GUARD_ITERATOR, (iterable -- iterable)) {
         bool definite = true;
         PyTypeObject *tp = sym_get_type(iterable);