[Mlir-commits] [mlir] [mlir][linalg] Expose getPreservedProducerResults method from ElementwiseOpFusion file (PR #73850)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Thu Dec 7 08:25:23 PST 2023
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 1ee6a1e38aa0c3773d892fcd01bb2af8e446e67f e21b3fb0dae35880d70cb86abbea33483fbc69db -- mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp
``````````
</details>
<details>
<summary>
View the diff from clang-format here.
</summary>
``````````diff
diff --git a/mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h b/mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
index fe3ccd1508..a848d12fbb 100644
--- a/mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
+++ b/mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
@@ -493,8 +493,8 @@ LogicalResult dropUnitDims(RewriterBase &rewriter, GenericOp genericOp,
struct ElementwiseOpFusionResult {
Operation *fusedOp;
llvm::DenseMap<Value, Value> replacements;
- static llvm::SmallDenseSet<int> getPreservedProducerResults(GenericOp producer,
- GenericOp consumer);
+ static llvm::SmallDenseSet<int>
+ getPreservedProducerResults(GenericOp producer, GenericOp consumer);
};
FailureOr<ElementwiseOpFusionResult>
fuseElementwiseOps(RewriterBase &rewriter, OpOperand *fusedOperand);
diff --git a/mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp b/mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp
index f296310f72..3eb9119075 100644
--- a/mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp
+++ b/mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp
@@ -75,7 +75,7 @@ static AffineMap getIndexingMapOfProducerOperandsInCoordinatesOfFusedOp(
/// be preserved after the fusion.
llvm::SmallDenseSet<int>
ElementwiseOpFusionResult::getPreservedProducerResults(GenericOp producer,
- GenericOp consumer) {
+ GenericOp consumer) {
llvm::SmallDenseSet<int> preservedProducerResults;
for (const auto &producerResult : llvm::enumerate(producer->getResults())) {
auto *outputOperand = producer.getDpsInitOperand(producerResult.index());
@@ -305,7 +305,8 @@ mlir::linalg::fuseElementwiseOps(RewriterBase &rewriter,
"expected producer of input operand");
/// Find the results of the producer that have uses outside of the consumer.
llvm::SmallDenseSet<int> preservedProducerResults =
- ElementwiseOpFusionResult::getPreservedProducerResults(producer, consumer);
+ ElementwiseOpFusionResult::getPreservedProducerResults(producer,
+ consumer);
// Compute the fused operands list and indexing maps.
SmallVector<Value> fusedInputOperands, fusedOutputOperands;
``````````
</details>
https://github.com/llvm/llvm-project/pull/73850
More information about the Mlir-commits
mailing list