[Mlir-commits] [mlir] 0c501db - [mlir][linalg] Make loop ops in TileLoopNest accessible

[Lei Zhang]

Author: Lei Zhang
Date: 2021-11-15T14:33:48-05:00
New Revision: 0c501db7d3c2b924ca8e68c981df10716ef59f6d

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

LOG: [mlir][linalg] Make loop ops in TileLoopNest accessible

Reviewed By: gysit

Differential Revision: https://reviews.llvm.org/D113927

Added: 
    

Modified: 
    mlir/include/mlir/Dialect/Linalg/Utils/Utils.h

Removed: 
    


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