diff --git a/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h b/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h --- a/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h +++ b/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h @@ -235,6 +235,9 @@ /// Returns the tiled root operation. LinalgOp getRootOp() { return rootOp; } + /// Returns the loop ops generated from tiling. + ArrayRef getLoopOps() { return tileLoopOps; } + private: /// Returns true if the tile loop nest has no tile loops. bool isEmpty();