[Mlir-commits] [mlir] [mlir][linalg] Use the tile size for untransformed Winograd axes (PR #226119)

[llvmlistbot at llvm.org]

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 origin/main HEAD --extensions cpp -- mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp --diff_from_common_commit
``````````

:warning:
The reproduction instructions above might return results for more than one PR
in a stack if you are using a stacked PR workflow. You can limit the results by
changing `origin/main` to the base branch/commit you want to compare against.
:warning:

</details>

<details>
<summary>
View the diff from clang-format here.
</summary>

``````````diff
diff --git a/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp b/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
index 29b9f6bde..601644499 100644
--- a/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
+++ b/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
@@ -3684,10 +3684,10 @@ WinogradInputTransformOp::getTiledImplementation(OpBuilder &builder,
   OpFoldResult offsetW = OpFoldResult(mappedOffsetW);
   sliceOffsets.append(
       {offsets[getOutputNDim()], offsetH, offsetW, offsets[getOutputCDim()]});
-  OpFoldResult sizeH = alphaH != 1 ? OpFoldResult(mappedSizeH)
-                                   : sizes[getOutputTileHDim()];
-  OpFoldResult sizeW = alphaW != 1 ? OpFoldResult(mappedSizeW)
-                                   : sizes[getOutputTileWDim()];
+  OpFoldResult sizeH =
+      alphaH != 1 ? OpFoldResult(mappedSizeH) : sizes[getOutputTileHDim()];
+  OpFoldResult sizeW =
+      alphaW != 1 ? OpFoldResult(mappedSizeW) : sizes[getOutputTileWDim()];
   sliceSizes.append(
       {sizes[getOutputNDim()], sizeH, sizeW, sizes[getOutputCDim()]});
   int64_t inputRank = getInputOperandRank();
@@ -3820,10 +3820,10 @@ LogicalResult WinogradOutputTransformOp::getResultTilePosition(
 
   OpFoldResult offsetH = OpFoldResult(mappedOffsetH);
   OpFoldResult offsetW = OpFoldResult(mappedOffsetW);
-  OpFoldResult sizeH = valueH != 1 ? OpFoldResult(mappedSizeH)
-                                   : sizes[getValueTileHDim()];
-  OpFoldResult sizeW = valueW != 1 ? OpFoldResult(mappedSizeW)
-                                   : sizes[getValueTileWDim()];
+  OpFoldResult sizeH =
+      valueH != 1 ? OpFoldResult(mappedSizeH) : sizes[getValueTileHDim()];
+  OpFoldResult sizeW =
+      valueW != 1 ? OpFoldResult(mappedSizeW) : sizes[getValueTileWDim()];
 
   resultOffsets.append(
       {offsets[getValueNDim()], offsetH, offsetW, offsets[getValueFDim()]});

``````````

</details>


https://github.com/llvm/llvm-project/pull/226119