more robust handling if measurement analysis results are not available

Author: johnzl-777

src/bloqade/stim/rewrite/ifs_to_stim.py

@@ -137,7 +137,7 @@ def rewrite_Statement(self, node: ir.Statement) -> RewriteResult:
 
     def rewrite_IfElse(self, stmt: scf.IfElse) -> RewriteResult:
 
-        condition_type = self.measure_frame.entries[stmt.cond]
+        condition_type = self.measure_frame.entries.get(stmt.cond)
         # Check the condition is a singular MeasurementIdBool and that
         # it was generated by querying if the measurement is equivalent to the one state
         if not isinstance(condition_type, MeasureIdBool):
@@ -162,9 +162,11 @@ def rewrite_IfElse(self, stmt: scf.IfElse) -> RewriteResult:
             return RewriteResult()
 
         # generate get record statement
-        measure_id_idx_stmt = py.Constant(
-            (condition_type.idx - 1) - self.measure_frame.num_measures_at_stmt[stmt]
-        )
+        num_measures = self.measure_frame.num_measures_at_stmt.get(stmt)
+        if num_measures is None:
+            return RewriteResult()
+
+        measure_id_idx_stmt = py.Constant((condition_type.idx - 1) - num_measures)
         get_record_stmt = GetRecord(id=measure_id_idx_stmt.result)  # noqa: F841
 
         address_attr = stmts[0].qubits.hints.get("address")