Forward defs to parent frame in inline-qasm

Author: weinbe58

src/bloqade/qasm2/dialects/inline.py

@@ -43,10 +43,12 @@ def lower(
 
         raw = textwrap.dedent(value)
         qasm_lowering = QASM2(main.union([ilist, glob, noise, parallel]))
-        region = qasm_lowering.run(loads(raw))
+        region, defs = qasm_lowering.run(loads(raw), return_defs=True)
         for qasm_stmt in region.blocks[0].stmts:
             qasm_stmt.detach()
             state.current_frame.push(qasm_stmt)
+            
+        state.current_frame.defs.update(defs)
 
         for block in region.blocks:
             for qasm_stmt in block.stmts:

src/bloqade/qasm2/parse/lowering.py

@@ -27,6 +27,7 @@ def run(
         lineno_offset: int = 0,
         col_offset: int = 0,
         compactify: bool = True,
+        return_defs: bool = False,
     ) -> ir.Region:
         # TODO: add source info
         state = lowering.State(
@@ -56,14 +57,23 @@ def run(
                 else:
                     e.args = (hint,)
                     raise e
-
+                
+            if return_defs:
+                defs = frame.defs.copy()
+                
             region = frame.curr_region
 
         if compactify:
             from kirin.rewrite import Walk, CFGCompactify
 
             Walk(CFGCompactify()).rewrite(region)
-        return region
+            
+        if return_defs:
+            # use this to update the current frame for python
+            # lowering for inline qasm.
+            return region, defs
+        else:
+            return region
 
     def visit(self, state: lowering.State[ast.Node], node: ast.Node) -> lowering.Result:
         name = node.__class__.__name__