feat: specialize isinstance for tuple of primitive types

Author: BobTheBuidler

mypyc/irbuild/specialize.py

@@ -30,6 +30,7 @@
     RefExpr,
     StrExpr,
     SuperExpr,
+    SymbolNode,
     TupleExpr,
     Var,
 )
@@ -653,13 +654,51 @@ def translate_isinstance(builder: IRBuilder, expr: CallExpr, callee: RefExpr) ->
             obj = builder.accept(obj_expr, can_borrow=can_borrow)
             return builder.builder.isinstance_helper(obj, irs, expr.line)
 
+    if isinstance(type_expr, TupleExpr) and type_expr.items:
+        nodes: list[SymbolNode | None] = []
+        for item in type_expr.items:
+            if not isinstance(item, RefExpr):
+                return None
+            if item.node is None:
+                return None
+            if item.node.fullname not in nodes:
+                nodes.append(item.node.fullname)
+
+        descs = [isinstance_primitives.get(fullname) for fullname in nodes]
+    
+        obj = builder.accept(expr.args[0])
+
+        retval = Register(bool_rprimitive)
+        pass_block = BasicBlock()
+        fail_block = BasicBlock()
+        exit_block = BasicBlock()
+
+        # Chain the checks: if any succeed, jump to pass_block; else, continue
+        for i, desc in enumerate(descs):
+            is_last = (i == len(descs) - 1)
+            next_block = fail_block if is_last else BasicBlock()
+            builder.add_bool_branch(builder.primitive_op(desc, [obj], expr.line), pass_block, next_block)
+            if not is_last:
+                builder.activate_block(next_block)
+
+        # If any check passed
+        builder.activate_block(pass_block)
+        builder.assign(retval, builder.true(), expr.line)
+        builder.goto(exit_block)
+
+        # If all checks failed
+        builder.activate_block(fail_block)
+        builder.assign(retval, builder.false(), expr.line)
+        builder.goto(exit_block)
+
+        # Return the result
+        builder.activate_block(exit_block)
+        return retval
+
     if isinstance(type_expr, RefExpr):
-        node = type_expr.node
-        if node:
-            desc = isinstance_primitives.get(node.fullname)
-            if desc:
-                obj = builder.accept(obj_expr)
-                return builder.primitive_op(desc, [obj], expr.line)
+        if node := type_expr.node:
+            if desc := isinstance_primitives.get(node.fullname):
+                return builder.primitive_op(desc, [builder.accept(obj_expr)], expr.line)
 
     elif isinstance(type_expr, TupleExpr):
         node_names: list[str] = []