MLIR
18.0.0git
|
Fuse two linalg.generic
operations that have a producer-consumer relationship captured through fusedOperand
.
More...
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
Static Public Member Functions | |
static llvm::SmallDenseSet< int > | getPreservedProducerResults (GenericOp producer, GenericOp consumer) |
Returns a set of indices of the producer's results which would be preserved after the fusion. More... | |
Public Attributes | |
Operation * | fusedOp |
llvm::DenseMap< Value, Value > | replacements |
Fuse two linalg.generic
operations that have a producer-consumer relationship captured through fusedOperand
.
The method expects that areElementwiseOpsFusable
returns true for the given fusedOperand
.
Definition at line 493 of file Transforms.h.
|
static |
Returns a set of indices of the producer's results which would be preserved after the fusion.
Definition at line 77 of file ElementwiseOpFusion.cpp.
References mlir::detail::enumerate().
Referenced by mlir::linalg::fuseElementwiseOps().
Operation* mlir::linalg::ElementwiseOpFusionResult::fusedOp |
Definition at line 494 of file Transforms.h.
Referenced by mlir::linalg::fuseElementwiseOps().
llvm::DenseMap<Value, Value> mlir::linalg::ElementwiseOpFusionResult::replacements |
Definition at line 495 of file Transforms.h.
Referenced by mlir::linalg::fuseElementwiseOps().