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

Classes
struct	Edge

struct	Node

Public Member Functions
	MemRefDependenceGraph (Block &block)

bool	init ()

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

References mlir::Value::getType().

◆ addNode()

unsigned MemRefDependenceGraph::addNode ( Operation * op )

Definition at line 374 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 692 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 705 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 259 of file Utils.h.

References print().

◆ forEachMemRefEdge()

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

Definition at line 729 of file Utils.cpp.

◆ forEachMemRefInputEdge()

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

Definition at line 713 of file Utils.cpp.

◆ forEachMemRefOutputEdge()

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

Definition at line 721 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 538 of file Utils.cpp.

◆ getForOpNode() [1/2]

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

inline

Definition at line 172 of file Utils.h.

References getForOpNode().

◆ getForOpNode() [2/2]

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

Definition at line 366 of file Utils.cpp.

Referenced by getForOpNode().

◆ getFusedLoopNestInsertionPoint()

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

Definition at line 552 of file Utils.cpp.

◆ getIncomingMemRefAccesses()

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

Definition at line 512 of file Utils.cpp.

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

◆ getNode() [1/2]

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

inline

Definition at line 162 of file Utils.h.

References getNode().

◆ getNode() [2/2]

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

Definition at line 359 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 528 of file Utils.cpp.

◆ hasDependencePath()

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

Definition at line 477 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 424 of file Utils.cpp.

◆ hasNode()

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

inline

Definition at line 168 of file Utils.h.

References nodes.

◆ init()

bool MemRefDependenceGraph::init ( )

Definition at line 246 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, mlir::affine::MemRefDependenceGraph::Node::op, and mlir::affine::MemRefDependenceGraph::Node::stores.

◆ print()

void MemRefDependenceGraph::print ( raw_ostream & os ) const

Definition at line 744 of file Utils.cpp.

Referenced by dump().

◆ removeEdge()

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

Definition at line 450 of file Utils.cpp.

References mlir::Value::getType().

◆ removeNode()

void MemRefDependenceGraph::removeNode ( unsigned id )

Definition at line 381 of file Utils.cpp.

◆ updateEdges() [1/2]

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

Definition at line 667 of file Utils.cpp.

◆ updateEdges() [2/2]

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

Definition at line 628 of file Utils.cpp.

◆ writesToLiveInOrEscapingMemrefs()

bool MemRefDependenceGraph::writesToLiveInOrEscapingMemrefs ( unsigned id ) const

Definition at line 404 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 262 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