[mlir][IR] Add additional rewriter constructor

[matthias-springer]

For convenience, add an additional constructor to RewriterBase and IRRewriter that also sets the insertion point. OpBuilder provides a similar constructor.

[llvmbot]

@llvm/pr-subscribers-mlir-core

@llvm/pr-subscribers-mlir

Author: Matthias Springer (matthias-springer)

Changes

For convenience, add an additional constructor to RewriterBase and IRRewriter that also sets the insertion point. OpBuilder provides a similar constructor.

---

Full diff: https://github.com/llvm/llvm-project/pull/85044.diff

1 Files Affected:

• (modified) mlir/include/mlir/IR/PatternMatch.h (+4)

diff --git a/mlir/include/mlir/IR/PatternMatch.h b/mlir/include/mlir/IR/PatternMatch.h
index ef53a0b82866fc..070e6ed702f86a 100644
--- a/mlir/include/mlir/IR/PatternMatch.h
+++ b/mlir/include/mlir/IR/PatternMatch.h
@@ -736,6 +736,8 @@ class RewriterBase : public OpBuilder {
       : OpBuilder(ctx, listener) {}
   explicit RewriterBase(const OpBuilder &otherBuilder)
       : OpBuilder(otherBuilder) {}
+  explicit RewriterBase(Operation *op, OpBuilder::Listener *listener = nullptr)
+      : OpBuilder(op, listener) {}
   virtual ~RewriterBase();
 
 private:
@@ -756,6 +758,8 @@ class IRRewriter : public RewriterBase {
   explicit IRRewriter(MLIRContext *ctx, OpBuilder::Listener *listener = nullptr)
       : RewriterBase(ctx, listener) {}
   explicit IRRewriter(const OpBuilder &builder) : RewriterBase(builder) {}
+  explicit IRRewriter(Operation *op, OpBuilder::Listener *listener = nullptr)
+      : RewriterBase(op, listener) {}
 };
 
 //===----------------------------------------------------------------------===//