diff --git a/mlir/include/mlir/Transforms/DialectConversion.h b/mlir/include/mlir/Transforms/DialectConversion.h index 536b23f5c33c1..5de58b89130a2 100644 --- a/mlir/include/mlir/Transforms/DialectConversion.h +++ b/mlir/include/mlir/Transforms/DialectConversion.h @@ -862,6 +862,9 @@ class ConversionPatternRewriter final : public PatternRewriter { /// Return a reference to the internal implementation. detail::ConversionPatternRewriterImpl &getImpl(); + /// Return "true" if the given operation was replaced or erased. + bool isOpIgnored(Operation *op) const; + private: // Allow OperationConverter to construct new rewriters. friend struct OperationConverter; diff --git a/mlir/lib/Transforms/Utils/DialectConversion.cpp b/mlir/lib/Transforms/Utils/DialectConversion.cpp index 7494ca9ec3784..f95340c718409 100644 --- a/mlir/lib/Transforms/Utils/DialectConversion.cpp +++ b/mlir/lib/Transforms/Utils/DialectConversion.cpp @@ -2245,6 +2245,10 @@ detail::ConversionPatternRewriterImpl &ConversionPatternRewriter::getImpl() { return *impl; } +bool ConversionPatternRewriter::isOpIgnored(Operation *op) const { + return getImpl()->isOpIgnored(op); +} + //===----------------------------------------------------------------------===// // ConversionPattern //===----------------------------------------------------------------------===//