#include "mlir/Dialect/Affine/Analysis/Utils.h"

Classes
struct	Edge

struct	Node

Public Member Functions
	MemRefDependenceGraph (Block &block)

bool	init (bool fullAffineDependences=true)

const Node *	getNode (unsigned id) const

Node *	getNode (unsigned id)

bool	hasNode (unsigned id) const

const Node *	getForOpNode (AffineForOp forOp) const

Node *	getForOpNode (AffineForOp forOp)

unsigned	addNode (Operation *op)

void	removeNode (unsigned id)

bool	writesToLiveInOrEscapingMemrefs (unsigned id) const

bool	hasEdge (unsigned srcId, unsigned dstId, Value value=nullptr) const

void	addEdge (unsigned srcId, unsigned dstId, Value value)

void	removeEdge (unsigned srcId, unsigned dstId, Value value)

bool	hasDependencePath (unsigned srcId, unsigned dstId) const

unsigned	getIncomingMemRefAccesses (unsigned id, Value memref) const

unsigned	getOutEdgeCount (unsigned id, Value memref=nullptr) const

void	gatherDefiningNodes (unsigned id, DenseSet< unsigned > &definingNodes) const
	Return all nodes which define SSA values used in node 'id'. More...

Operation *	getFusedLoopNestInsertionPoint (unsigned srcId, unsigned dstId) const

void	updateEdges (unsigned srcId, unsigned dstId, const DenseSet< Value > &privateMemRefs, bool removeSrcId)

void	updateEdges (unsigned sibId, unsigned dstId)

void	addToNode (unsigned id, ArrayRef< Operation * > loads, ArrayRef< Operation * > stores, ArrayRef< Operation * > memrefLoads, ArrayRef< Operation * > memrefStores, ArrayRef< Operation * > memrefFrees)

void	clearNodeLoadAndStores (unsigned id)

void	forEachMemRefInputEdge (unsigned id, const std::function< void(Edge)> &callback)

void	forEachMemRefOutputEdge (unsigned id, const std::function< void(Edge)> &callback)

void	forEachMemRefEdge (ArrayRef< Edge > edges, const std::function< void(Edge)> &callback)

void	print (raw_ostream &os) const

void	dump () const

Public Attributes
DenseMap< unsigned, Node >	nodes

DenseMap< unsigned, SmallVector< Edge, 2 > >	inEdges

DenseMap< unsigned, SmallVector< Edge, 2 > >	outEdges

DenseMap< Value, unsigned >	memrefEdgeCount

unsigned	nextNodeId = 0

Block &	block
	The block for which this graph is created to perform fusion. More...

Detailed Description

Definition at line 62 of file Utils.h.

Constructor & Destructor Documentation

◆ MemRefDependenceGraph()

mlir::affine::MemRefDependenceGraph::MemRefDependenceGraph ( Block & block )

inline

Definition at line 154 of file Utils.h.

Member Function Documentation

◆ addEdge()

void MemRefDependenceGraph::addEdge	(	unsigned	srcId,
		unsigned	dstId,
		Value	value
	)

Definition at line 532 of file Utils.cpp.

References mlir::Value::getType().

◆ addNode()

unsigned MemRefDependenceGraph::addNode ( Operation * op )

Definition at line 467 of file Utils.cpp.

References mlir::affine::MemRefDependenceGraph::Node::id.

◆ addToNode()

void MemRefDependenceGraph::addToNode	(	unsigned	id,
		ArrayRef< Operation * >	loads,
		ArrayRef< Operation * >	stores,
		ArrayRef< Operation * >	memrefLoads,
		ArrayRef< Operation * >	memrefStores,
		ArrayRef< Operation * >	memrefFrees
	)

Definition at line 784 of file Utils.cpp.

References mlir::affine::MemRefDependenceGraph::Node::loads, mlir::affine::MemRefDependenceGraph::Node::memrefFrees, mlir::affine::MemRefDependenceGraph::Node::memrefLoads, mlir::affine::MemRefDependenceGraph::Node::memrefStores, and mlir::affine::MemRefDependenceGraph::Node::stores.

◆ clearNodeLoadAndStores()

void MemRefDependenceGraph::clearNodeLoadAndStores ( unsigned id )

Definition at line 797 of file Utils.cpp.

References mlir::affine::MemRefDependenceGraph::Node::loads, and mlir::affine::MemRefDependenceGraph::Node::stores.

◆ dump()

void mlir::affine::MemRefDependenceGraph::dump ( ) const

inline

Definition at line 267 of file Utils.h.

References print().

◆ forEachMemRefEdge()

void MemRefDependenceGraph::forEachMemRefEdge	(	ArrayRef< Edge >	edges,
		const std::function< void(Edge)> &	callback
	)

Definition at line 821 of file Utils.cpp.

◆ forEachMemRefInputEdge()

void MemRefDependenceGraph::forEachMemRefInputEdge	(	unsigned	id,
		const std::function< void(Edge)> &	callback
	)

Definition at line 805 of file Utils.cpp.

◆ forEachMemRefOutputEdge()

void MemRefDependenceGraph::forEachMemRefOutputEdge	(	unsigned	id,
		const std::function< void(Edge)> &	callback
	)

Definition at line 813 of file Utils.cpp.

◆ gatherDefiningNodes()

void MemRefDependenceGraph::gatherDefiningNodes	(	unsigned	id,
		DenseSet< unsigned > &	definingNodes
	)		const

Return all nodes which define SSA values used in node 'id'.

Definition at line 631 of file Utils.cpp.

◆ getForOpNode() [1/2]

Node* mlir::affine::MemRefDependenceGraph::getForOpNode ( AffineForOp forOp )

inline

Definition at line 180 of file Utils.h.

References getForOpNode().

◆ getForOpNode() [2/2]

const Node * MemRefDependenceGraph::getForOpNode ( AffineForOp forOp ) const

Definition at line 459 of file Utils.cpp.

Referenced by getForOpNode().

◆ getFusedLoopNestInsertionPoint()

Operation * MemRefDependenceGraph::getFusedLoopNestInsertionPoint	(	unsigned	srcId,
		unsigned	dstId
	)		const

Definition at line 645 of file Utils.cpp.

◆ getIncomingMemRefAccesses()

unsigned MemRefDependenceGraph::getIncomingMemRefAccesses	(	unsigned	id,
		Value	memref
	)		const

Definition at line 605 of file Utils.cpp.

References mlir::affine::MemRefDependenceGraph::Node::getStoreOpCount().

◆ getNode() [1/2]

Node* mlir::affine::MemRefDependenceGraph::getNode ( unsigned id )

inline

Definition at line 170 of file Utils.h.

References getNode().

◆ getNode() [2/2]

const Node * MemRefDependenceGraph::getNode ( unsigned id ) const

Definition at line 452 of file Utils.cpp.

Referenced by canRemoveSrcNodeAfterFusion(), gatherEscapingMemrefs(), gatherProducerConsumerMemrefs(), getNode(), and getProducerCandidates().

◆ getOutEdgeCount()

unsigned MemRefDependenceGraph::getOutEdgeCount	(	unsigned	id,
		Value	memref = `nullptr`
	)		const

Definition at line 621 of file Utils.cpp.

◆ hasDependencePath()

bool MemRefDependenceGraph::hasDependencePath	(	unsigned	srcId,
		unsigned	dstId
	)		const

Definition at line 570 of file Utils.cpp.

References mlir::affine::MemRefDependenceGraph::Edge::id, and mlir::Operation::isBeforeInBlock().

◆ hasEdge()

bool MemRefDependenceGraph::hasEdge	(	unsigned	srcId,
		unsigned	dstId,
		Value	value = `nullptr`
	)		const

Definition at line 517 of file Utils.cpp.

◆ hasNode()

bool mlir::affine::MemRefDependenceGraph::hasNode ( unsigned id ) const

inline

Definition at line 176 of file Utils.h.

References nodes.

◆ init()

bool MemRefDependenceGraph::init ( bool fullAffineDependences = true )

Definition at line 336 of file Utils.cpp.

References addNodeToMDG(), mlir::affine::getAffineForIVs(), mlir::Operation::getResults(), mlir::Operation::getUsers(), mlir::affine::MemRefDependenceGraph::Node::hasFree(), mlir::affine::MemRefDependenceGraph::Node::hasStore(), mlir::affine::MemRefDependenceGraph::Node::id, mlir::isMemoryEffectFree(), mlir::affine::MemRefDependenceGraph::Node::loads, mayDependence(), mlir::affine::MemRefDependenceGraph::Node::op, and mlir::affine::MemRefDependenceGraph::Node::stores.

◆ print()

void MemRefDependenceGraph::print ( raw_ostream & os ) const

Definition at line 833 of file Utils.cpp.

Referenced by dump().

◆ removeEdge()

void MemRefDependenceGraph::removeEdge	(	unsigned	srcId,
		unsigned	dstId,
		Value	value
	)

Definition at line 543 of file Utils.cpp.

References mlir::Value::getType().

◆ removeNode()

void MemRefDependenceGraph::removeNode ( unsigned id )

Definition at line 474 of file Utils.cpp.

◆ updateEdges() [1/2]

void MemRefDependenceGraph::updateEdges	(	unsigned	sibId,
		unsigned	dstId
	)

Definition at line 759 of file Utils.cpp.

◆ updateEdges() [2/2]

void MemRefDependenceGraph::updateEdges	(	unsigned	srcId,
		unsigned	dstId,
		const DenseSet< Value > &	privateMemRefs,
		bool	removeSrcId
	)

Definition at line 720 of file Utils.cpp.

◆ writesToLiveInOrEscapingMemrefs()

bool MemRefDependenceGraph::writesToLiveInOrEscapingMemrefs ( unsigned id ) const

Definition at line 497 of file Utils.cpp.

References mlir::affine::MemRefDependenceGraph::Node::stores.

Member Data Documentation

◆ block

Block& mlir::affine::MemRefDependenceGraph::block

The block for which this graph is created to perform fusion.

Definition at line 270 of file Utils.h.

Referenced by gatherEscapingMemrefs().

◆ inEdges

DenseMap<unsigned, SmallVector<Edge, 2> > mlir::affine::MemRefDependenceGraph::inEdges

Definition at line 144 of file Utils.h.

Referenced by getProducerCandidates().

◆ memrefEdgeCount

DenseMap<Value, unsigned> mlir::affine::MemRefDependenceGraph::memrefEdgeCount

Definition at line 150 of file Utils.h.

◆ nextNodeId

unsigned mlir::affine::MemRefDependenceGraph::nextNodeId = 0

Definition at line 152 of file Utils.h.

Referenced by addNodeToMDG().

◆ nodes

DenseMap<unsigned, Node> mlir::affine::MemRefDependenceGraph::nodes

Definition at line 141 of file Utils.h.

Referenced by addNodeToMDG(), and hasNode().

◆ outEdges

DenseMap<unsigned, SmallVector<Edge, 2> > mlir::affine::MemRefDependenceGraph::outEdges

Definition at line 147 of file Utils.h.

Referenced by canRemoveSrcNodeAfterFusion().

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

include/mlir/Dialect/Affine/Analysis/Utils.h
lib/Dialect/Affine/Analysis/Utils.cpp

Classes

Public Member Functions

Public Attributes

Detailed Description

Constructor & Destructor Documentation

◆ MemRefDependenceGraph()

Member Function Documentation

◆ addEdge()

◆ addNode()

◆ addToNode()

◆ clearNodeLoadAndStores()

◆ dump()

◆ forEachMemRefEdge()

◆ forEachMemRefInputEdge()

◆ forEachMemRefOutputEdge()

◆ gatherDefiningNodes()

◆ getForOpNode() [1/2]

◆ getForOpNode() [2/2]

◆ getFusedLoopNestInsertionPoint()

◆ getIncomingMemRefAccesses()

◆ getNode() [1/2]

◆ getNode() [2/2]

◆ getOutEdgeCount()

◆ hasDependencePath()

◆ hasEdge()

◆ hasNode()

◆ init()

◆ print()

◆ removeEdge()

◆ removeNode()

◆ updateEdges() [1/2]

◆ updateEdges() [2/2]

◆ writesToLiveInOrEscapingMemrefs()

Member Data Documentation

◆ block

◆ inEdges

◆ memrefEdgeCount

◆ nextNodeId

◆ nodes

◆ outEdges