[Mlir-commits] [mlir] [mlir][tensor] Update `GeneralizeOuterUnitDimsPackOpPattern` (PR #115312)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Thu Nov 7 05:06:49 PST 2024
github-actions[bot] wrote:
<!--LLVM CODE FORMAT COMMENT: {clang-format}-->
:warning: C/C++ code formatter, clang-format found issues in your code. :warning:
<details>
<summary>
You can test this locally with the following command:
</summary>
``````````bash
git-clang-format --diff 2d7f34f2a5df9396a33a0ea044cfe3ddf33e1f5c 8ed63a25bc7f9b4c49bd920cd6407be7147c3b2f --extensions h,cpp -- mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp
``````````
</details>
<details>
<summary>
View the diff from clang-format here.
</summary>
``````````diff
diff --git a/mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp b/mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp
index 0be8799f32..f7409f75b9 100644
--- a/mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp
+++ b/mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp
@@ -1196,20 +1196,22 @@ LogicalResult GeneralizeOuterUnitDimsPackOpPattern::matchAndRewrite(
srcPermForTranspose.append(SmallVector<int64_t>(packOp.getInnerDimsPos()));
LLVM_DEBUG(DBGS() << "Pack permutation: " << packOp << "\n";
- llvm::interleaveComma(srcPermForTranspose, DBGS() << "perm: "); DBGSNL(););
+ llvm::interleaveComma(srcPermForTranspose, DBGS() << "perm: ");
+ DBGSNL(););
// 2.1 Create tensor.empty (init value for TransposeOp)
SmallVector<OpFoldResult> transShapeForEmptyOp(srcRank - numTiles,
oneIdxAttr);
transShapeForEmptyOp.append(tileSizes);
- applyPermutationToVector<OpFoldResult>(transShapeForEmptyOp, srcPermForTranspose);
+ applyPermutationToVector<OpFoldResult>(transShapeForEmptyOp,
+ srcPermForTranspose);
Value empty = rewriter.create<tensor::EmptyOp>(
loc, transShapeForEmptyOp, packOp.getSourceType().getElementType());
// 2.2 Create linalg.transpose
- auto transposedOp =
- rewriter.create<linalg::TransposeOp>(loc, input, empty, srcPermForTranspose);
+ auto transposedOp = rewriter.create<linalg::TransposeOp>(loc, input, empty,
+ srcPermForTranspose);
// 3. Insert the inner tile to the destination:
// %inserted_tile = tensor.insert_slice(%transposed_tile)
``````````
</details>
https://github.com/llvm/llvm-project/pull/115312
More information about the Mlir-commits
mailing list