fix lowering

Author: Roger-luo

src/bloqade/qasm2/dialects/expr/lowering.py

@@ -36,7 +36,7 @@ def lower_Assign(self, state: lowering.State, node: ast.Assign):
                 raise lowering.BuildError(f"unsupported target syntax {target_syntax}")
 
     def lower_Expr(self, state: lowering.State, node: ast.Expr):
-        return state.lower(node.value)
+        return state.parent.visit(state, node.value)
 
     def lower_Constant(self, state: lowering.State, node: ast.Constant):
         if isinstance(node.value, int):

src/bloqade/qasm2/dialects/inline.py

@@ -30,15 +30,16 @@ def lower(
         if isinstance(text, ast.Constant) and isinstance(text.value, str):
             value = text.value
         elif isinstance(text, ast.Name) and isinstance(text.ctx, ast.Load):
-            value = state.get_global(text.id).expect(str)
+            value = state.get_global(text).expect(str)
         else:
             raise lowering.BuildError(
                 "InlineQASM takes a string literal or global string"
             )
 
         raw = textwrap.dedent(value)
         qasm_lowering = QASM2(state.parent.dialects)
-        qasm_lowering.run(loads(raw))
+        code = qasm_lowering.run(loads(raw))
+        code.print()
 
 
 # NOTE: this is a dummy statement that won't appear in IR.

src/bloqade/qasm2/parse/lowering.py

@@ -21,6 +21,7 @@ def run(
         self,
         stmt: ast.Node,
         *,
+        state: lowering.State | None = None,
         source: str | None = None,
         globals: dict[str, Any] | None = None,
         file: str | None = None,
@@ -29,7 +30,7 @@ def run(
         compactify: bool = True,
     ) -> ir.Statement:
         # TODO: add source info
-        state = lowering.State(
+        state = state or lowering.State(
             self,
             file=file,
             lineno_offset=lineno_offset,

test/qasm2/test_inline.py

@@ -88,3 +88,27 @@ def qasm2_inline_code():
         qasm2.inline(lines)
 
     qasm2_inline_code.print()
+
+
+lines = textwrap.dedent(
+    """
+OPENQASM 2.0;
+
+qreg q[2];
+creg c[2];
+
+h q[0];
+CX q[0], q[1];
+barrier q[0], q[1];
+CX q[0], q[1];
+rx(pi/2) q[0];
+"""
+)
+
+
+@qasm2.main.add(inline)
+def qasm2_inline_code():
+    qasm2.inline(lines)
+
+
+qasm2_inline_code.print()