pr: Fix assign result conversion

trailofbits · Jan 29, 2025 · 202c344 · 202c344
1 parent 27d3596
commit 202c344
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/vast/Conversion/Parser/ToParser.cpp b/lib/vast/Conversion/Parser/ToParser.cpp
@@ -566,8 +566,12 @@ namespace vast::conv {
             ) const override {
                 auto args = realized_operand_values(adaptor.getOperands(), rewriter);
                 rewriter.create< pr::Assign >(op.getLoc(), std::vector< mlir_type >(), args);
-                rewriter.replaceAllUsesWith(op, args[0]);
-                rewriter.eraseOp(op);
+
+                if (!op->getUsers().empty()) {
+                    rewriter.replaceOp(op, args[0]);
+                } else {
+                    rewriter.eraseOp(op);
+                }
                 return mlir::success();
             }
         };