[Mlir-commits] [mlir] c1e9131 - [mlir][transform] Expose transform op from TrackingListener

[Matthias Springer]

Author: Matthias Springer
Date: 2023-05-24T15:11:24+02:00
New Revision: c1e9131d9b27f5746c8ba685c17afbcc9e409c92

URL: https://github.com/llvm/llvm-project/commit/c1e9131d9b27f5746c8ba685c17afbcc9e409c92
DIFF: https://github.com/llvm/llvm-project/commit/c1e9131d9b27f5746c8ba685c17afbcc9e409c92.diff

LOG: [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

Added: 
    

Modified: 
    mlir/include/mlir/Dialect/Transform/IR/TransformOps.h

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Dialect/Transform/IR/TransformOps.h b/mlir/include/mlir/Dialect/Transform/IR/TransformOps.h
index 543eba9df7ab..f77982f1790c 100644
--- a/mlir/include/mlir/Dialect/Transform/IR/TransformOps.h
+++ b/mlir/include/mlir/Dialect/Transform/IR/TransformOps.h
@@ -68,6 +68,9 @@ class TrackingListener : public RewriterBase::Listener,
   /// 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;