Replace op->erase() with rewriter.eraseOp (#1009)

Ref: https://mlir.llvm.org/docs/PatternRewriter/#pattern-rewriter
libxsmm · Feb 9, 2025 · 32acfc9 · 32acfc9
1 parent e9cee09
commit 32acfc9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/TPP/Transforms/HoistVectorTransfers.cpp b/lib/TPP/Transforms/HoistVectorTransfers.cpp
@@ -226,10 +226,10 @@ struct HoistVectorTransferOp : OpRewritePattern<vector::ContractionOp> {
     // Erase the old vector contract operation
     for (auto result : contractOp->getResults()) {
       for (auto *userOp : result.getUsers()) {
-        userOp->erase();
+        rewriter.eraseOp(userOp);
       }
     }
-    contractOp.erase();
+    rewriter.eraseOp(contractOp);
 
     return success();
   }