fixing test adding fixedpoint to unroll

Author: weinbe58

src/bloqade/cirq_utils/emit/base.py

@@ -2,16 +2,15 @@
 from warnings import warn
 from dataclasses import field, dataclass
 
-from bloqade.rewrite.passes import AggressiveUnroll
 import cirq
-from kirin import ir, types, interp, passes
+from kirin import ir, types, interp
 from kirin.emit import EmitABC, EmitError, EmitFrame
 from kirin.interp import MethodTable, impl
-from kirin.passes import inline
-from kirin.dialects import func, py
+from kirin.dialects import py, func
 from typing_extensions import Self
 
 from bloqade.squin import kernel
+from bloqade.rewrite.passes import AggressiveUnroll
 
 
 def emit_circuit(
@@ -115,21 +114,17 @@ def main():
 
     emitter = EmitCirq(qubits=circuit_qubits)
 
-
-
     symbol_op_trait = mt.code.get_trait(ir.SymbolOpInterface)
     if (symbol_op_trait := mt.code.get_trait(ir.SymbolOpInterface)) is None:
-        raise EmitError(
-            f"The method is not a symbol, cannot emit circuit!"
-        )
+        raise EmitError("The method is not a symbol, cannot emit circuit!")
 
     sym_name = symbol_op_trait.get_sym_name(mt.code).unwrap()
 
     if (signature_trait := mt.code.get_trait(ir.HasSignature)) is None:
         raise EmitError(
             f"The method {sym_name} does not have a signature, cannot emit circuit!"
         )
-    
+
     signature = signature_trait.get_signature(mt.code)
     new_signature = func.Signature(inputs=(), output=signature.output)
 
@@ -149,14 +144,13 @@ def main():
         arg_ssa.replace_by(value.result)
         entry_block.args.delete(arg_ssa)
 
-    new_func = func.Function(sym_name=sym_name, body=callable_region, signature=new_signature)
+    new_func = func.Function(
+        sym_name=sym_name, body=callable_region, signature=new_signature
+    )
     mt_ = ir.Method(None, None, sym_name, [], mt.dialects, new_func)
 
-    passes.Fold(mt_.dialects, no_raise=False)(mt_)
+    AggressiveUnroll(mt_.dialects).fixpoint(mt_)
     mt_.print(hint="const")
-
-    # AggressiveUnroll(mt_.dialects)(mt_)
-    # mt_.print(hint="const")
     return emitter.run(mt_, args=())
 
 

test/pyqrack/squin/test_noise.py

@@ -8,6 +8,8 @@ def main():
         q = squin.qalloc(1)
         squin.qubit_loss(1.0, q[0])
 
+        return q
+
     target = PyQrack(1)
     result = target.run(main)