Skip to content

Commit 437fcc6

Browse files
[mlir][IR] Add additional rewriter constructor (llvm#85044)
For convenience, add an additional constructor to `RewriterBase` and `IRRewriter` that also sets the insertion point. `OpBuilder` provides a similar constructor.
1 parent 1e9bfcd commit 437fcc6

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

mlir/include/mlir/IR/PatternMatch.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -736,6 +736,8 @@ class RewriterBase : public OpBuilder {
736736
: OpBuilder(ctx, listener) {}
737737
explicit RewriterBase(const OpBuilder &otherBuilder)
738738
: OpBuilder(otherBuilder) {}
739+
explicit RewriterBase(Operation *op, OpBuilder::Listener *listener = nullptr)
740+
: OpBuilder(op, listener) {}
739741
virtual ~RewriterBase();
740742

741743
private:
@@ -756,6 +758,8 @@ class IRRewriter : public RewriterBase {
756758
explicit IRRewriter(MLIRContext *ctx, OpBuilder::Listener *listener = nullptr)
757759
: RewriterBase(ctx, listener) {}
758760
explicit IRRewriter(const OpBuilder &builder) : RewriterBase(builder) {}
761+
explicit IRRewriter(Operation *op, OpBuilder::Listener *listener = nullptr)
762+
: RewriterBase(op, listener) {}
759763
};
760764

761765
//===----------------------------------------------------------------------===//

0 commit comments

Comments
 (0)