diff --git a/mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransforms.h b/mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransforms.h --- a/mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransforms.h +++ b/mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransforms.h @@ -63,6 +63,9 @@ ArrayRef sizes, StringRef linalgMarker, ArrayRef permutation); + +/// Tiles ops similar to `tileLinalgOpAndSetMarker` but generates loop.parallel +/// operations instead. LogicalResult tileLinalgOpToParallelLoopsAndSetMarker( PatternRewriter &rewriter, Operation *op, ArrayRef sizes, StringRef linalgMarker, ArrayRef permutation); @@ -72,6 +75,9 @@ LogicalResult tileAndFuseLinalgOpAndSetMarker( PatternRewriter &rewriter, Operation *op, ArrayRef sizes, ArrayRef operandIndicesToFuse, StringRef linalgMarker); + +/// Tiles ops similar to `tileAndFuseLinalgOpAndSetMarker` but generates +/// loop.parallel operations instead. LogicalResult tileAndFuseLinalgOpToParallelLoopsAndSetMarker( PatternRewriter &rewriter, Operation *op, ArrayRef sizes, ArrayRef operandIndicesToFuse, StringRef linalgMarker);