mlir::linalg::ElementwiseOpFusionResult Struct Reference

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

Detailed Description

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 498 of file Transforms.h.

Member Function Documentation

getPreservedProducerResults()

llvm::SmallDenseSet< int > ElementwiseOpFusionResult::getPreservedProducerResults ( GenericOp  producer, GenericOp  consumer  )

static

Returns a set of indices of the producer's results which would be preserved after the fusion.

Definition at line 76 of file ElementwiseOpFusion.cpp.

References mlir::detail::enumerate().

Referenced by mlir::linalg::fuseElementwiseOps().

Member Data Documentation

fusedOp

Operation* mlir::linalg::ElementwiseOpFusionResult::fusedOp

Definition at line 499 of file Transforms.h.

Referenced by mlir::linalg::fuseElementwiseOps().

replacements

llvm::DenseMap<Value, Value> mlir::linalg::ElementwiseOpFusionResult::replacements

Definition at line 500 of file Transforms.h.

Referenced by mlir::linalg::fuseElementwiseOps().

---

The documentation for this struct was generated from the following files:

• include/mlir/Dialect/Linalg/Transforms/Transforms.h
• lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp