From c1e9131d9b27f5746c8ba685c17afbcc9e409c92 Mon Sep 17 00:00:00 2001 From: Matthias Springer Date: Wed, 24 May 2023 15:02:56 +0200 Subject: [PATCH] [mlir][transform] Expose transform op from TrackingListener This allows subclasses (such as the ErrorCheckingTrackingListener in IREE) to produce better error messages. Differential Revision: https://reviews.llvm.org/D151322 --- mlir/include/mlir/Dialect/Transform/IR/TransformOps.h | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mlir/include/mlir/Dialect/Transform/IR/TransformOps.h b/mlir/include/mlir/Dialect/Transform/IR/TransformOps.h index 543eba9..f77982f 100644 --- a/mlir/include/mlir/Dialect/Transform/IR/TransformOps.h +++ b/mlir/include/mlir/Dialect/Transform/IR/TransformOps.h @@ -68,6 +68,9 @@ protected: /// Return the single op that defines all given values (if any). static Operation *getCommonDefiningOp(ValueRange values); + /// Return the transform op in which this TrackingListener is used. + TransformOpInterface getTransformOp() const { return transformOp; } + private: void notifyOperationRemoved(Operation *op) override; -- 2.7.4