pr: Remove noparse arguments from noparse when folding.

trailofbits · Jan 8, 2025 · bd57a54 · bd57a54
1 parent 6f03642
commit bd57a54
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/lib/vast/Dialect/Parser/Ops.cpp b/lib/vast/Dialect/Parser/Ops.cpp
@@ -18,7 +18,17 @@ namespace vast::pr {
     using fold_result_t = ::llvm::SmallVectorImpl< ::mlir::OpFoldResult >;
 
     logical_result NoParse::fold(FoldAdaptor adaptor, fold_result_t &results) {
-        return mlir::failure();
+        auto change = mlir::failure();
+        auto op = getOperation();
+
+        for (auto [idx, operand] : llvm::reverse(llvm::enumerate(getOperands()))) {
+            if (auto noparse = mlir::dyn_cast< NoParse >(operand.getDefiningOp())) {
+                op->eraseOperand(idx);
+                change = mlir::success();
+            }
+        }
+
+        return change;
     }
 
 } // namespace vast::pr