diff --git a/src/bloqade/rewrite/passes/aggressive_unroll.py b/src/bloqade/rewrite/passes/aggressive_unroll.py index 989eac70..eadeb56a 100644 --- a/src/bloqade/rewrite/passes/aggressive_unroll.py +++ b/src/bloqade/rewrite/passes/aggressive_unroll.py @@ -42,6 +42,7 @@ def unsafe_run(self, mt: Method) -> RewriteResult: ilist.rewrite.InlineGetItem(), ilist.rewrite.FlattenAdd(), ilist.rewrite.HintLen(), + DeadCodeElimination(), ) result = Fixpoint(Walk(rule)).rewrite(mt.code).join(result) diff --git a/src/bloqade/rewrite/passes/canonicalize_ilist.py b/src/bloqade/rewrite/passes/canonicalize_ilist.py index 4514fec0..88b42adb 100644 --- a/src/bloqade/rewrite/passes/canonicalize_ilist.py +++ b/src/bloqade/rewrite/passes/canonicalize_ilist.py @@ -5,6 +5,7 @@ Walk, Chain, Fixpoint, + DeadCodeElimination, ) from kirin.dialects.ilist import rewrite @@ -24,6 +25,7 @@ def unsafe_run(self, mt: ir.Method): rewrite.InlineGetItem(), rewrite.FlattenAdd(), rewrite.HintLen(), + DeadCodeElimination(), ) ) ).rewrite(mt.code)