[Mlir-commits] [mlir] [MLIR][SCF] Fold dim ops of iter_args to respective init_args (PR #109973)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Wed Sep 25 05:48:42 PDT 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 40d6497a97a61ef5ea38e615cc324104cb34072a 3dfdef6e8319601a6169fefe23636412d8f31a55 --extensions cpp -- mlir/lib/Dialect/SCF/IR/SCF.cpp
``````````
</details>
<details>
<summary>
View the diff from clang-format here.
</summary>
``````````diff
diff --git a/mlir/lib/Dialect/SCF/IR/SCF.cpp b/mlir/lib/Dialect/SCF/IR/SCF.cpp
index 62c78f134d..9705a9cd17 100644
--- a/mlir/lib/Dialect/SCF/IR/SCF.cpp
+++ b/mlir/lib/Dialect/SCF/IR/SCF.cpp
@@ -1478,7 +1478,6 @@ struct DimOfForallOp : public OpRewritePattern<tensor::DimOp> {
}
};
-
/// Fold dim ops of iter_args to dim ops of their respective init args. E.g.:
///
/// ```
@@ -1506,7 +1505,8 @@ struct DimOfForallIterArg : public OpRewritePattern<tensor::DimOp> {
auto blockArg = dyn_cast<BlockArgument>(dimOp.getSource());
if (!blockArg)
return failure();
- auto forallOp = dyn_cast<ForallOp>(blockArg.getParentBlock()->getParentOp());
+ auto forallOp =
+ dyn_cast<ForallOp>(blockArg.getParentBlock()->getParentOp());
if (!forallOp)
return failure();
Value initArg = forallOp.getTiedLoopInit(blockArg)->get();
@@ -1890,9 +1890,10 @@ struct FoldTensorCastOfOutputIntoForallOp
void ForallOp::getCanonicalizationPatterns(RewritePatternSet &results,
MLIRContext *context) {
- results.add<DimOfForallOp, DimOfForallIterArg, FoldTensorCastOfOutputIntoForallOp,
- ForallOpControlOperandsFolder, ForallOpIterArgsFolder,
- ForallOpSingleOrZeroIterationDimsFolder>(context);
+ results.add<DimOfForallOp, DimOfForallIterArg,
+ FoldTensorCastOfOutputIntoForallOp, ForallOpControlOperandsFolder,
+ ForallOpIterArgsFolder, ForallOpSingleOrZeroIterationDimsFolder>(
+ context);
}
/// Given the region at `index`, or the parent operation if `index` is None,
``````````
</details>
https://github.com/llvm/llvm-project/pull/109973
More information about the Mlir-commits
mailing list