optimize MATCH_SEQUENCE/MAPPING

Author: Sacul0457

Lib/test/test_capi/test_opt.py

@@ -4684,6 +4684,41 @@ class A:
         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.assertIn("_MATCH_MAPPING", uops)
+        self.assertNotIn("_GUARD_BIT_IS_SET_POP", 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.assertIn("_MATCH_SEQUENCE", uops)
+        self.assertNotIn("_GUARD_BIT_IS_SET_POP", uops)
+
     def test_dict_update(self):
         def testfunc(n):
             d = {1: 2, 3: 4}

Python/optimizer_bytecodes.c

@@ -2071,6 +2071,22 @@ dummy_func(void) {
         n = names;
     }
 
+    op(_MATCH_MAPPING, (subject -- subject, res)) {
+        if (sym_has_type(subject)) {
+            PyTypeObject *type = sym_get_type(subject);
+            int match = type->tp_flags & Py_TPFLAGS_MAPPING;
+            res = match ? sym_new_const(ctx, Py_True) : sym_new_const(ctx, Py_False);
+        }
+    }
+
+    op(_MATCH_SEQUENCE, (subject -- subject, res)) {
+        if (sym_has_type(subject)) {
+            PyTypeObject *type = sym_get_type(subject);
+            int match = type->tp_flags & Py_TPFLAGS_SEQUENCE;
+            res = match ? sym_new_const(ctx, Py_True) : sym_new_const(ctx, Py_False);
+        }
+    }
+
     op(_DICT_UPDATE, (dict, unused[oparg - 1], update -- dict, unused[oparg - 1], upd)) {
         (void)dict;
         upd = update;