Skip to content

Commit 6de4093

Browse files
authored
Add missing .join(result) in GlobalToParallel pass (#293)
1 parent 0737238 commit 6de4093

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

src/bloqade/qasm2/passes/glob.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -58,13 +58,15 @@ def unsafe_run(self, mt: ir.Method) -> abc.RewriteResult:
5858
rewriter = walk.Walk(self.generate_rule(mt))
5959
result = rewriter.rewrite(mt.code)
6060

61-
result = walk.Walk(dce.DeadCodeElimination()).rewrite(mt.code)
62-
result = Fixpoint(walk.Walk(rule=cse.CommonSubexpressionElimination())).rewrite(
63-
mt.code
61+
result = walk.Walk(dce.DeadCodeElimination()).rewrite(mt.code).join(result)
62+
result = (
63+
Fixpoint(walk.Walk(rule=cse.CommonSubexpressionElimination()))
64+
.rewrite(mt.code)
65+
.join(result)
6466
)
6567

6668
# do fold again to get proper hint for inserted const
67-
result = Fold(mt.dialects)(mt)
69+
result = Fold(mt.dialects)(mt).join(result)
6870
return result
6971

7072

@@ -110,10 +112,12 @@ def unsafe_run(self, mt: ir.Method) -> abc.RewriteResult:
110112
rewriter = walk.Walk(self.generate_rule(mt))
111113
result = rewriter.rewrite(mt.code)
112114

113-
result = walk.Walk(dce.DeadCodeElimination()).rewrite(mt.code)
114-
result = Fixpoint(walk.Walk(rule=cse.CommonSubexpressionElimination())).rewrite(
115-
mt.code
115+
result = walk.Walk(dce.DeadCodeElimination()).rewrite(mt.code).join(result)
116+
result = (
117+
Fixpoint(walk.Walk(rule=cse.CommonSubexpressionElimination()))
118+
.rewrite(mt.code)
119+
.join(result)
116120
)
117121
# do fold again to get proper hint
118-
result = Fold(mt.dialects)(mt)
122+
result = Fold(mt.dialects)(mt).join(result)
119123
return result

0 commit comments

Comments
 (0)