[PATCH] D78140: [mlir][Linalg] NFC: Add missing comments from earlier commit (D77965)
Mahesh Ravishankar via Phabricator via llvm-commits
llvm-commits at lists.llvm.org
Tue Apr 14 12:24:44 PDT 2020
mravishankar updated this revision to Diff 257442.
mravishankar added a comment.
Address comment.
Repository:
rG LLVM Github Monorepo
CHANGES SINCE LAST ACTION
https://reviews.llvm.org/D78140/new/
https://reviews.llvm.org/D78140
Files:
mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransforms.h
Index: mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransforms.h
===================================================================
--- mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransforms.h
+++ mlir/include/mlir/Dialect/Linalg/Transforms/LinalgTransforms.h
@@ -49,10 +49,10 @@
// success.
////////////////////////////////////////////////////////////////////////////////
-/// Tiles `op` by `sizes` permuting the loops according to `permutation`
-/// and sets the attribute `kLinalgTransformMarker` to `linalgMarker`.
-/// The permutation is expressed as a list of integers that specify
-/// the new ordering of the loop nest. The length of `permutation`
+/// Tiles `op` by `sizes` permuting the loops according to `permutation` and
+/// sets the attribute `kLinalgTransformMarker` to `linalgMarker`. The
+/// permutation is expressed as a list of integers that specify the new ordering
+/// of the loop nest (using loop.for operations). The length of `permutation`
/// must be equal to the length of `tileSizes`.
/// E.g. the permutation `(i,j,k) -> (j,k,i)` will be expressed with
/// `permutation = [1,2,0]`. All values in `permutation` must be
@@ -63,6 +63,9 @@
ArrayRef<int64_t> sizes,
StringRef linalgMarker,
ArrayRef<unsigned> permutation);
+
+/// Tiles ops similar to `tileLinalgOpAndSetMarker` but generates loop.parallel
+/// operations instead.
LogicalResult tileLinalgOpToParallelLoopsAndSetMarker(
PatternRewriter &rewriter, Operation *op, ArrayRef<int64_t> sizes,
StringRef linalgMarker, ArrayRef<unsigned> permutation);
@@ -72,6 +75,9 @@
LogicalResult tileAndFuseLinalgOpAndSetMarker(
PatternRewriter &rewriter, Operation *op, ArrayRef<int64_t> sizes,
ArrayRef<int64_t> operandIndicesToFuse, StringRef linalgMarker);
+
+/// Tiles ops similar to `tileAndFuseLinalgOpAndSetMarker` but generates
+/// loop.parallel operations instead.
LogicalResult tileAndFuseLinalgOpToParallelLoopsAndSetMarker(
PatternRewriter &rewriter, Operation *op, ArrayRef<int64_t> sizes,
ArrayRef<int64_t> operandIndicesToFuse, StringRef linalgMarker);
-------------- next part --------------
A non-text attachment was scrubbed...
Name: D78140.257442.patch
Type: text/x-patch
Size: 2215 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20200414/98c42fd9/attachment.bin>
More information about the llvm-commits
mailing list