gh-131798: Add MATCH_SEQUENCE/MAPPING to the JIT optimizer

Lib/test/test_capi/test_opt.py

@@ -5710,10 +5710,42 @@ class A:
         res, ex = self._run_with_optimizer(testfunc, TIER2_THRESHOLD)
         self.assertEqual(res, TIER2_THRESHOLD)
         uops = get_opnames(ex)
-
         self.assertIn("_MATCH_CLASS", uops)
         self.assertEqual(count_ops(ex, "_POP_TOP_NOP"), 4)
 
+    def test_match_mapping(self):
+        def testfunc(n):
+            x = {}
+            ret = 0
+            for _ in range(n):
+                x["a"] = 1
+                match x:
+                    case {}:
+                        ret += 1
+            return ret
+
+        res, ex = self._run_with_optimizer(testfunc, TIER2_THRESHOLD)
+        self.assertEqual(res, TIER2_THRESHOLD)
+        uops = get_opnames(ex)
+        self.assertNotIn("_MATCH_MAPPING", uops)
+        self.assertIn("_LOAD_COMMON_CONSTANT", uops)
+
+    def test_match_sequence(self):
+        def testfunc(n):
+            ret = 0
+            for _ in range(n):
+                x = 1, 2
+                match x:
+                    case a, b:
+                        ret += 1
+            return ret
+
+        res, ex = self._run_with_optimizer(testfunc, TIER2_THRESHOLD)
+        self.assertEqual(res, TIER2_THRESHOLD)
+        uops = get_opnames(ex)
+        self.assertNotIn("_MATCH_SEQUENCE", uops)
+        self.assertIn("_LOAD_COMMON_CONSTANT", uops)
+
     def test_dict_update(self):
         def testfunc(n):
             d = {1: 2, 3: 4}

Python/optimizer_bytecodes.c

@@ -2595,6 +2595,30 @@ dummy_func(void) {
         n = names;
     }
 
+    op(_MATCH_MAPPING, (subject -- subject, res)) {
+        PyTypeObject *type = sym_get_type(subject);
+        if (type != NULL) {
+            int match = type->tp_flags & Py_TPFLAGS_MAPPING;
+            REPLACE_OP(this_instr, _LOAD_COMMON_CONSTANT, match? CONSTANT_TRUE : CONSTANT_FALSE, 0);
+            res = sym_new_const(ctx, match? Py_True : Py_False);
+        }
+        else {
+            res = sym_new_type(ctx, &PyBool_Type);
+        }
+    }
+
+    op(_MATCH_SEQUENCE, (subject -- subject, res)) {
+        PyTypeObject *type = sym_get_type(subject);
+        if (type != NULL) {
+            int match = type->tp_flags & Py_TPFLAGS_SEQUENCE;
+            REPLACE_OP(this_instr, _LOAD_COMMON_CONSTANT, match? CONSTANT_TRUE : CONSTANT_FALSE, 0);
+            res = sym_new_const(ctx, match? Py_True : Py_False);
+        }
+        else {
+            res = sym_new_type(ctx, &PyBool_Type);
+        }
+    }
+
     op(_DICT_UPDATE, (dict, unused[oparg - 1], update -- dict, unused[oparg - 1], upd)) {
         (void)dict;
         upd = update;