This is a utility class for mapping one set of IR entities to another. More...
#include "mlir/IR/IRMapping.h"
Public Member Functions | |
| void | map (Value from, Value to) |
| Inserts a new mapping for 'from' to 'to'. | |
| void | map (Block *from, Block *to) |
| void | map (Operation *from, Operation *to) |
| template<typename S, typename T, std::enable_if_t<!std::is_assignable_v< Value, S > &&!std::is_assignable_v< Block *, S > &&!std::is_assignable_v< Operation *, S > > * = nullptr> | |
| void | map (S &&from, T &&to) |
| template<typename T> | |
| void | erase (T from) |
| Erases a mapping for 'from'. | |
| template<typename T> | |
| bool | contains (T from) const |
| Checks to see if a mapping for 'from' exists. | |
| template<typename T> | |
| auto | lookupOrNull (T from) const |
| Lookup a mapped value within the map. | |
| template<typename T> | |
| auto | lookupOrDefault (T from) const |
| Lookup a mapped value within the map. | |
| template<typename T> | |
| auto | lookup (T from) const |
| Lookup a mapped value within the map. | |
| void | clear () |
| Clears all mappings held by the mapper. | |
| const DenseMap< Value, Value > & | getValueMap () const |
| Return the held value mapping. | |
| const DenseMap< Block *, Block * > & | getBlockMap () const |
| Return the held block mapping. | |
| const DenseMap< Operation *, Operation * > & | getOperationMap () const |
| Return the held operation mapping. | |
Detailed Description
This is a utility class for mapping one set of IR entities to another.
New mappings can be inserted via 'map'. Existing mappings can be found via the 'lookup*' functions. There are three variants that differ only in return value when an existing is not found for the provided key: SSA values, blocks, and operations. 'lookupOrNull' returns nullptr where as 'lookupOrDefault' will return the lookup key.
Definition at line 26 of file IRMapping.h.
Member Function Documentation
◆ clear()
|
inline |
Clears all mappings held by the mapper.
Definition at line 79 of file IRMapping.h.
Referenced by addExecuteResults(), and hoistAffineIfOp().
◆ contains()
|
inline |
Checks to see if a mapping for 'from' exists.
Definition at line 51 of file IRMapping.h.
Referenced by mlir::Region::cloneInto(), mlir::generateUnrolledLoop(), and mlir::shard::partitionOperation().
◆ erase()
|
inline |
Erases a mapping for 'from'.
Definition at line 45 of file IRMapping.h.
◆ getBlockMap()
Return the held block mapping.
Definition at line 85 of file IRMapping.h.
◆ getOperationMap()
Return the held operation mapping.
Definition at line 88 of file IRMapping.h.
◆ getValueMap()
Return the held value mapping.
Definition at line 82 of file IRMapping.h.
◆ lookup()
|
inline |
Lookup a mapped value within the map.
This asserts the provided value exists within the map.
Definition at line 72 of file IRMapping.h.
References lookupOrNull(), and result.
Referenced by buildPackingLoopNestImpl(), calculateGatherOffset(), mlir::Region::cloneInto(), mlir::OpBuilder::cloneRegionBefore(), mlir::linalg::createAllReduceForResultsWithoutPartialShardings(), mlir::linalg::createDestinationPassingStyleInitOperands(), generateUnrolledLoop(), mlir::generateUnrolledLoop(), handleArgumentImpl(), mlir::ReductionNode::initialize(), outlineKernelFuncImpl(), mlir::linalg::partitionLinalgOpWithShardedReduction(), mlir::shard::partitionOperation(), reduceIfNeeded(), remainsLegalAfterInline(), replaceByPackingResult(), rewriteOneForallCommonImpl(), vectorizeLinalgYield(), vectorizeOneOp(), and vectorizeTensorExtract().
◆ lookupOrDefault()
|
inline |
Lookup a mapped value within the map.
If a mapping for the provided value does not exist then return the provided value.
Definition at line 65 of file IRMapping.h.
Referenced by buildPackingLoopNestImpl(), mlir::Operation::clone(), mlir::fuseIndependentSiblingForLoops(), generateFusedElementwiseOpRegion(), mlir::generateUnrolledLoop(), haveNoReadsAfterWriteExceptSameIndex(), inlinePayload(), inlineRegionAndEmitStore(), processParallelLoop(), and vectorizeUniform().
◆ lookupOrNull()
|
inline |
Lookup a mapped value within the map.
If a mapping for the provided value does not exist then return nullptr.
Definition at line 58 of file IRMapping.h.
Referenced by mlir::affine::fuseLoops(), lookup(), remapInlinedOperands(), and vectorizeOperand().
◆ map() [1/4]
Definition at line 31 of file IRMapping.h.
◆ map() [2/4]
Definition at line 32 of file IRMapping.h.
◆ map() [3/4]
|
inline |
Definition at line 38 of file IRMapping.h.
References map().
◆ map() [4/4]
Inserts a new mapping for 'from' to 'to'.
If there is an existing mapping, it is overwritten.
Definition at line 30 of file IRMapping.h.
Referenced by mlir::sparse_tensor::Merger::buildExp(), buildPackingLoopNestImpl(), mlir::Operation::clone(), mlir::Region::cloneInto(), mlir::linalg::createAllReduceForResultsWithoutPartialShardings(), mlir::emitc::createExpression(), createFullTiles(), createParallelComputeFunction(), mlir::query::extractFunction(), fuseIfLegal(), mlir::fuseIndependentSiblingForallLoops(), mlir::fuseIndependentSiblingForLoops(), genCoIterateBranchNest(), generateFusedElementwiseOpRegion(), generateShiftedLoop(), generateUnrolledLoop(), mlir::generateUnrolledLoop(), mlir::impl::OutlineShapeComputationPassBase< DerivedT >::getDependentDialects(), handleArgumentImpl(), injectGpuIndexOperations(), mlir::inlineCall(), inlinePayload(), inlineRegionAndEmitStore(), mlir::makeRegionIsolatedFromAbove(), map(), CanonicalizeContractAdd< AddOpType >::matchAndRewrite(), outlineExecuteOp(), outlineKernelFuncImpl(), mlir::outlineSingleBlockRegion(), mlir::shard::partitionBlock(), mlir::linalg::partitionLinalgOpWithShardedReduction(), mlir::shard::partitionOperation(), mlir::shard::partitionOperation(), processParallelLoop(), rewriteOneForallCommonImpl(), specializeParallelLoopForUnrolling(), tileAndFuseFirstExtractUseThroughContainingOpBlockArgument(), vectorizeAsLinalgGeneric(), and verifyDependencies().
The documentation for this class was generated from the following file:
- include/mlir/IR/IRMapping.h
Generated on for MLIR by
1.14.0