[Mlir-commits] [mlir] [MLIR][Shard] Fold all_slice(all_gather(...)) pairs (PR #193906)

[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/Shard/Transforms/Simplify.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/Shard/Transforms/Simplify.cpp b/mlir/lib/Dialect/Shard/Transforms/Simplify.cpp
index 904080cf2..5a188a517 100644
--- a/mlir/lib/Dialect/Shard/Transforms/Simplify.cpp
+++ b/mlir/lib/Dialect/Shard/Transforms/Simplify.cpp
@@ -132,8 +132,8 @@ struct AllReduceAllSliceSimplification : OpRewritePattern<AllSliceOp> {
 };
 
 // Simplify AllSliceOp(AllGatherOp) -> input when both ops share the same grid,
-// grid_axes and axis. all_gather replicates grouped slices along gather_axis and
-// all_slice immediately picks the per-rank slice back out on the same axis.
+// grid_axes and axis. all_gather replicates grouped slices along gather_axis
+// and all_slice immediately picks the per-rank slice back out on the same axis.
 struct AllGatherAllSliceSimplification : OpRewritePattern<AllSliceOp> {
   using OpRewritePattern::OpRewritePattern;
 

``````````

</details>


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