#include "mlir/Dialect/Affine/Analysis/Utils.h"#include "mlir/Analysis/Presburger/PresburgerRelation.h"#include "mlir/Dialect/Affine/Analysis/AffineAnalysis.h"#include "mlir/Dialect/Affine/Analysis/LoopAnalysis.h"#include "mlir/Dialect/Affine/IR/AffineOps.h"#include "mlir/Dialect/Affine/IR/AffineValueMap.h"#include "mlir/Dialect/MemRef/IR/MemRef.h"#include "mlir/Dialect/Utils/StaticValueUtils.h"#include "mlir/IR/IntegerSet.h"#include "llvm/ADT/SetVector.h"#include "llvm/ADT/SmallVectorExtras.h"#include "llvm/Support/Debug.h"#include "llvm/Support/DebugLog.h"#include "llvm/Support/raw_ostream.h"#include <optional>Go to the source code of this file.
Macros | |
| #define | DEBUG_TYPE "analysis-utils" |
Typedefs | |
| using | Node = MemRefDependenceGraph::Node |
Functions | |
| template<typename... EffectTys> | |
| static void | getEffectedValues (Operation *op, SmallVectorImpl< Value > &values) |
| Returns the values that this op has a memref effect of type EffectTys on, not considering recursive effects. | |
| static Node * | addNodeToMDG (Operation *nodeOp, MemRefDependenceGraph &mdg, DenseMap< Value, SetVector< unsigned > > &memrefAccesses) |
| Add op to MDG creating a new node and adding its memory accesses (affine or non-affine to memrefAccesses (memref -> list of nodes with accesses) map. | |
| static Value | getMemRef (Operation *memOp) |
| Returns the memref being read/written by a memref/affine load/store op. | |
| static bool | mayDependence (const Node &srcNode, const Node &dstNode, Value memref) |
| Returns true if there may be a dependence on memref from srcNode's memory ops to dstNode's memory ops, while using the affine memory dependence analysis checks. | |
| static void | findInstPosition (Operation *op, Block *limitBlock, SmallVectorImpl< unsigned > *positions) |
| static Operation * | getInstAtPosition (ArrayRef< unsigned > positions, unsigned level, Block *block) |
| static LogicalResult | addMissingLoopIVBounds (SmallPtrSet< Value, 8 > &ivs, FlatAffineValueConstraints *cst) |
| static std::optional< uint64_t > | getConstDifference (AffineMap lbMap, AffineMap ubMap) |
| static std::optional< int64_t > | getMemoryFootprintBytes (Block &block, Block::iterator start, Block::iterator end, int memorySpace) |
| static void | unpackOptionalValues (ArrayRef< std::optional< Value > > source, SmallVector< Value > &target) |
| static LogicalResult | alignAndAddBound (FlatAffineValueConstraints &constraints, BoundType type, unsigned pos, AffineMap map, ValueRange operands) |
| Bound an identifier pos in a given FlatAffineValueConstraints with constraints drawn from an affine map. | |
| static AffineMap | addConstToResults (AffineMap map, int64_t val) |
| Add val to each result of map. | |
Variables | |
| const char *const | kSliceFusionBarrierAttrName = "slice_fusion_barrier" |
Macro Definition Documentation
◆ DEBUG_TYPE
Typedef Documentation
◆ Node
| using Node = MemRefDependenceGraph::Node |
Function Documentation
◆ addConstToResults()
Add val to each result of map.
Definition at line 2237 of file Utils.cpp.
References mlir::AffineMap::get(), mlir::AffineMap::getContext(), mlir::AffineMap::getNumDims(), mlir::AffineMap::getNumSymbols(), and mlir::AffineMap::getResults().
Referenced by mlir::affine::simplifyConstrainedMinMaxOp().
◆ addMissingLoopIVBounds()
|
static |
Definition at line 1582 of file Utils.cpp.
References mlir::affine::FlatAffineValueConstraints::addAffineForOpDomain(), mlir::affine::getForInductionVarOwner(), mlir::presburger::IntegerRelation::getNumDimVars(), mlir::FlatLinearValueConstraints::getValue(), mlir::affine::isAffineForInductionVar(), and success().
Referenced by mlir::affine::computeSliceUnion().
◆ addNodeToMDG()
|
static |
Add op to MDG creating a new node and adding its memory accesses (affine or non-affine to memrefAccesses (memref -> list of nodes with accesses) map.
Definition at line 192 of file Utils.cpp.
References mlir::affine::LoopNestStateCollector::collect(), getEffectedValues(), mlir::affine::LoopNestStateCollector::loadOpInsts, mlir::affine::LoopNestStateCollector::memrefFrees, mlir::affine::LoopNestStateCollector::memrefLoads, mlir::affine::LoopNestStateCollector::memrefStores, mlir::affine::MemRefDependenceGraph::nextNodeId, mlir::affine::MemRefDependenceGraph::nodes, mlir::affine::LoopNestStateCollector::storeOpInsts, and ValueRange.
Referenced by mlir::affine::MemRefDependenceGraph::init().
◆ alignAndAddBound()
|
static |
Bound an identifier pos in a given FlatAffineValueConstraints with constraints drawn from an affine map.
Before adding the constraint, the dimensions/symbols of the affine map are aligned with constraints. operands are the SSA Value operands used with the affine map. Note: This function adds a new symbol column to the constraints for each dimension/symbol that exists in the affine map but not in constraints.
Definition at line 2222 of file Utils.cpp.
References mlir::affine::FlatAffineValueConstraints::addBound(), mlir::alignAffineMapWithValues(), mlir::FlatLinearValueConstraints::appendSymbolVar(), mlir::FlatLinearValueConstraints::getMaybeValues(), and unpackOptionalValues().
Referenced by mlir::affine::simplifyConstrainedMinMaxOp().
◆ findInstPosition()
|
static |
Definition at line 1540 of file Utils.cpp.
References mlir::Block::begin(), mlir::Operation::getBlock(), and mlir::Block::getParentOp().
Referenced by mlir::affine::insertBackwardComputationSlice().
◆ getConstDifference()
Definition at line 1795 of file Utils.cpp.
References mlir::AffineMap::getNumDims(), mlir::AffineMap::getNumResults(), mlir::AffineMap::getNumSymbols(), mlir::AffineMap::getResult(), and mlir::simplifyAffineExpr().
Referenced by mlir::affine::buildSliceTripCountMap().
◆ getEffectedValues()
|
static |
Returns the values that this op has a memref effect of type EffectTys on, not considering recursive effects.
Definition at line 166 of file Utils.cpp.
References mlir::Operation::getOperands(), mlir::Value::getType(), and mlir::Operation::hasTrait().
Referenced by addNodeToMDG().
◆ getInstAtPosition()
|
static |
Definition at line 1557 of file Utils.cpp.
References b, getInstAtPosition(), and mlir::Operation::getRegions().
Referenced by getInstAtPosition(), and mlir::affine::insertBackwardComputationSlice().
◆ getMemoryFootprintBytes()
|
static |
Definition at line 2121 of file Utils.cpp.
References mlir::WalkResult::advance(), mlir::Block::begin(), mlir::Operation::emitError(), mlir::Operation::emitWarning(), mlir::Operation::getLoc(), mlir::affine::getNestingDepth(), inserted(), result, and mlir::Block::walk().
◆ getMemRef()
Returns the memref being read/written by a memref/affine load/store op.
Definition at line 247 of file Utils.cpp.
Referenced by mlir::affine::MemRefDependenceGraph::Node::getLoadAndStoreMemrefSet(), mlir::affine::MemRefDependenceGraph::Node::getLoadOpsForMemref(), getProducerCandidates(), mlir::affine::MemRefDependenceGraph::Node::getStoreOpsForMemref(), and mayDependence().
◆ mayDependence()
Returns true if there may be a dependence on memref from srcNode's memory ops to dstNode's memory ops, while using the affine memory dependence analysis checks.
The method assumes that there is at least one memory op in srcNode's loads and stores on memref, and similarly for dstNode. srcNode.op and destNode.op are expected to be nested in the same block and so the dependences are tested at the depth of that block.
Definition at line 265 of file Utils.cpp.
References mlir::affine::checkMemrefAccessDependence(), getMemRef(), mlir::affine::getNestingDepth(), mlir::affine::MemRefAccess::memref, and mlir::affine::noDependence().
Referenced by mlir::affine::MemRefDependenceGraph::init().
◆ unpackOptionalValues()
|
static |
Definition at line 2209 of file Utils.cpp.
References target.
Referenced by alignAndAddBound(), and mlir::affine::simplifyConstrainedMinMaxOp().
Variable Documentation
◆ kSliceFusionBarrierAttrName
| const char* const kSliceFusionBarrierAttrName = "slice_fusion_barrier" |
Definition at line 1862 of file Utils.cpp.
Referenced by mlir::affine::getComputationSliceState().
Generated on for MLIR by
1.14.0