debugging empty registry problem

Author: johnzl-777

squin_op_playground.py

@@ -40,11 +40,13 @@ def as_int(value: int):
 fold_pass = Fold(squin_with_qasm_core)
 fold_pass(constructed_method)
 
-frame, _ = address.AddressAnalysis(constructed_method.dialects).run_analysis(
+address_frame, _ = address.AddressAnalysis(constructed_method.dialects).run_analysis(
     constructed_method, no_raise=False
 )
 
-frame, _ = shape.ShapeAnalysis(constructed_method.dialects).run_analysis(
+constructed_method.print(analysis=address_frame.entries)
+
+shape_frame, _ = shape.ShapeAnalysis(constructed_method.dialects).run_analysis(
     constructed_method, no_raise=False
 )
 
@@ -53,4 +55,4 @@ def as_int(value: int):
     constructed_method, no_raise=False
 """
 
-constructed_method.print(analysis=frame.entries)
+constructed_method.print(analysis=shape_frame.entries)

src/bloqade/analysis/address/analysis.py

@@ -22,6 +22,7 @@ class AddressAnalysis(Forward[Address]):
     def initialize(self):
         super().initialize()
         self.next_address: int = 0
+        print(self.registry.statements)
         return self
 
     @property

src/bloqade/squin/analysis/shape/analysis.py

@@ -1,6 +1,6 @@
 # from typing import cast
 
-from kirin import ir, interp
+from kirin import ir
 from kirin.analysis import Forward
 from kirin.analysis.forward import ForwardFrame
 
@@ -17,12 +17,17 @@ class ShapeAnalysis(Forward[Shape]):
 
     def initialize(self):
         super().initialize()
+        print(self.registry.statements)
         return self
 
     # Take a page from const prop in Kirin,
     # I can get the data I want from the SizedTrait
     # and go from there
+
+    ## This gets called before the registry look up
     def eval_stmt(self, frame: ForwardFrame, stmt: ir.Statement):
+        # something fishy, registry pops up empty?
+        # This doesn't happen with the
         method = self.lookup_registry(frame, stmt)
         if method is not None:
             return method(self, frame, stmt)
@@ -37,9 +42,10 @@ def eval_stmt(self, frame: ForwardFrame, stmt: ir.Statement):
         else:
             return (NoShape(),)
 
+    # For when no implementation is found for the statement
     def eval_stmt_fallback(
         self, frame: ForwardFrame[Shape], stmt: ir.Statement
-    ) -> tuple[Shape, ...] | interp.SpecialValue[Shape]:
+    ) -> tuple[Shape, ...]:  # some form of Shape will go back into the frame
         return tuple(
             (
                 self.lattice.top()