update to new kirin

Author: johnzl-777

src/bloqade/analysis/record/analysis.py

@@ -42,15 +42,19 @@ class RecordAnalysis(ForwardExtra[RecordFrame, Record]):
     keys = ["record"]
     lattice = Record
 
-    def initialize_frame(self, code, *, has_parent_access: bool = False) -> RecordFrame:
-        return RecordFrame(code, has_parent_access=has_parent_access)
+    def initialize_frame(
+        self, node: ir.Statement, *, has_parent_access: bool = False
+    ) -> RecordFrame:
+        return RecordFrame(node, has_parent_access=has_parent_access)
 
-    def eval_stmt_fallback(self, frame: RecordFrame, stmt) -> tuple[Record, ...]:
-        return tuple(self.lattice.bottom() for _ in stmt.results)
+    def eval_stmt_fallback(
+        self, frame: RecordFrame, node: ir.Statement
+    ) -> tuple[Record, ...]:
+        return tuple(self.lattice.bottom() for _ in node.results)
 
     def run_method(self, method, args: tuple[Record, ...]):
         # NOTE: we do not support dynamic calls here, thus no need to propagate method object
-        return self.run_callable(method.code, (self.lattice.bottom(),) + args)
+        return self.run_method(method.code, (self.lattice.bottom(),) + args)
 
     T = TypeVar("T")
 
@@ -63,3 +67,6 @@ def get_const_value(
                 return hint.data
 
         return None
+
+    def method_self(self, method: ir.Method) -> Record:
+        return self.lattice.bottom()

test/analysis/record/test_record_analysis.py

@@ -1,10 +1,8 @@
 from bloqade import squin
-
-# from bloqade.analysis.record import RecordAnalysis
-# from bloqade.stim.passes.soft_flatten import SoftFlatten
+from bloqade.analysis.record import RecordAnalysis
+from bloqade.stim.passes.soft_flatten import SoftFlatten
 
 """
-
 @squin.kernel
 def test():
     qs = squin.qalloc(5)
@@ -49,18 +47,18 @@ def analysis_demo():
     qs = squin.qalloc(3)
     ms0 = squin.broadcast.measure(qs)
     ms1 = squin.broadcast.measure(qs)
-    squin.set_detector(ms0, coordinates=(0, 0))
-    squin.set_detector(ms1, coordinates=(0, 1))
+    squin.set_detector(ms0, coordinates=[0, 0])
+    squin.set_detector(ms1, coordinates=[0, 1])
     squin.broadcast.measure(qs)
-    squin.set_detector(ms1, coordinates=(0, 2))
+    squin.set_detector(ms1, coordinates=[0, 2])
 
     # get aliasing to work
     ms1 = ms0
-    squin.set_detector(ms1, coordinates=(1, 0))
+    squin.set_detector(ms1, coordinates=[1, 0])
     # return ms1
 
 
-# SoftFlatten(dialects=analysis_demo.dialects).fixpoint(analysis_demo)
-# analysis_demo.print()
-# frame, _ = RecordAnalysis(dialects=analysis_demo.dialects).run_analysis(analysis_demo)
-# analysis_demo.print(analysis=frame.entries)
+SoftFlatten(dialects=analysis_demo.dialects).fixpoint(analysis_demo)
+analysis_demo.print()
+frame, _ = RecordAnalysis(dialects=analysis_demo.dialects).run(analysis_demo)
+analysis_demo.print(analysis=frame.entries)