#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/Utils/StaticValueUtils.h"
#include "mlir/IR/IntegerSet.h"
#include "llvm/ADT/SetVector.h"
#include "llvm/Support/Debug.h"
#include "llvm/Support/DebugLog.h"
#include "llvm/Support/raw_ostream.h"
#include <optional>

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. More...

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. More...

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. More...

static AffineMap	addConstToResults (AffineMap map, int64_t val)
	Add `val` to each result of `map`. More...

Variables
const char *const	kSliceFusionBarrierAttrName = "slice_fusion_barrier"

Macro Definition Documentation

◆ DEBUG_TYPE

#define DEBUG_TYPE "analysis-utils"

Definition at line 29 of file Utils.cpp.

Typedef Documentation

◆ Node

using Node = MemRefDependenceGraph::Node

Definition at line 37 of file Utils.cpp.

Function Documentation

◆ addConstToResults()

static AffineMap addConstToResults	(	AffineMap	map,
		int64_t	val
	)

static

Add val to each result of map.

Definition at line 2145 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 LogicalResult addMissingLoopIVBounds	(	SmallPtrSet< Value, 8 > &	ivs,
		FlatAffineValueConstraints *	cst
	)

static

Definition at line 1489 of file Utils.cpp.

References mlir::affine::FlatAffineValueConstraints::addAffineForOpDomain(), mlir::remark::failed(), mlir::affine::getForInductionVarOwner(), mlir::presburger::IntegerRelation::getNumDimVars(), mlir::FlatLinearValueConstraints::getValue(), and mlir::affine::isAffineForInductionVar().

Referenced by mlir::affine::computeSliceUnion().

◆ addNodeToMDG()

static Node* addNodeToMDG	(	Operation *	nodeOp,
		MemRefDependenceGraph &	mdg,
		DenseMap< Value, SetVector< unsigned >> &	memrefAccesses
	)

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 190 of file Utils.cpp.

References mlir::affine::LoopNestStateCollector::collect(), mlir::affine::LoopNestStateCollector::loadOpInsts, mlir::affine::MemRefDependenceGraph::Node::loads, mlir::affine::LoopNestStateCollector::memrefFrees, mlir::affine::LoopNestStateCollector::memrefLoads, mlir::affine::LoopNestStateCollector::memrefStores, mlir::affine::MemRefDependenceGraph::nextNodeId, mlir::affine::MemRefDependenceGraph::nodes, and mlir::affine::LoopNestStateCollector::storeOpInsts.

Referenced by mlir::affine::MemRefDependenceGraph::init().

◆ alignAndAddBound()

static LogicalResult alignAndAddBound	(	FlatAffineValueConstraints &	constraints,
		BoundType	type,
		unsigned	pos,
		AffineMap	map,
		ValueRange	operands
	)

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 2130 of file Utils.cpp.

References mlir::affine::FlatAffineValueConstraints::addBound(), mlir::alignAffineMapWithValues(), mlir::FlatLinearValueConstraints::appendSymbolVar(), mlir::FlatLinearValueConstraints::getMaybeValues(), and unpackOptionalValues().

Referenced by mlir::affine::simplifyConstrainedMinMaxOp().