#include <stmt.h>

Inheritance diagram for freetensor::MatMulNode:

Collaboration diagram for freetensor::MatMulNode:

Public Member Functions
std::vector< Stmt >	children () const override

void	compHash () override

	DEFINE_NODE_TRAIT (MatMul)

Public Member Functions inherited from freetensor::StmtNode
void	setId (const ID &id=ID::make())

ID	id () const

const Metadata &	metadata () const

Metadata &	metadata ()

bool	isStmt () const override

virtual bool	isCtrlFlow () const

virtual std::vector< Ref< StmtNode > >	children () const

Ref< StmtNode >	prevLeafStmtInDFSOrder () const

Ref< StmtNode >	nextLeafStmtInDFSOrder () const

Ref< StmtNode >	prevStmtInDFSPostOrder () const

Ref< StmtNode >	nextStmtInDFSPreOrder () const

Ref< StmtNode >	ancestorById (const ID &lookup) const

bool	isAncestorOf (const Stmt &other) const

bool	isBefore (const Stmt &other) const

	DEFINE_NODE_ACCESS (Stmt)

Ref< StmtNode >	parentStmt () const

Ref< StmtNode >	parentStmtByFilter (const std::function< bool(const Stmt &)> &filter) const

Ref< StmtNode >	prevStmt () const

Ref< StmtNode >	nextStmt () const

Ref< StmtNode >	parentCtrlFlow () const

Ref< StmtNode >	prevInCtrlFlow () const

Ref< StmtNode >	nextInCtrlFlow () const

Public Member Functions inherited from freetensor::ASTNode
virtual	~ASTNode ()

virtual ASTNodeType	nodeType () const =0

bool	isAST () const override

virtual bool	isFunc () const

virtual bool	isStmt () const

virtual bool	isExpr () const

Ref< ASTNode >	parentAST () const

std::source_location	debugBlame () const

void	setDebugBlame (std::source_location loc)

	DEFINE_NODE_ACCESS (AST)

Public Member Functions inherited from freetensor::ASTPart
virtual	~ASTPart ()

	ASTPart (ASTPart &&other)

	ASTPart (const ASTPart &other)

ASTPart &	operator= (ASTPart &&)

ASTPart &	operator= (const ASTPart &)

bool	trySetParent (const Ref< ASTPart > &parent)

void	resetParent ()

Ref< ASTPart >	parent () const

bool	isSubTree () const

int	depth () const

virtual void	modifiedHook ()

size_t	hash ()

virtual bool	isAST () const

Public Member Functions inherited from freetensor::EnableSelf< ASTPart >
Ref< ASTPart >	self () const

Public Attributes
MatMulBackend	backend_

SubTree< CutlassMicroKernelProperty, NullPolicy::Nullable >	cutlassMicroKernelProperty_ = ChildOf{this}

SubTree< ExprNode >	a_ = ChildOf{this}

SubTree< ExprNode >	b_ = ChildOf{this}

SubTree< ExprNode >	c_ = ChildOf{this}

SubTree< ExprNode >	alpha_ = ChildOf{this}

SubTree< ExprNode >	beta_ = ChildOf{this}

SubTree< ExprNode >	m_ = ChildOf{this}

SubTree< ExprNode >	k_ = ChildOf{this}

SubTree< ExprNode >	n_ = ChildOf{this}

SubTree< ExprNode >	lda_ = ChildOf{this}

SubTree< ExprNode >	ldb_ = ChildOf{this}

SubTree< ExprNode >	ldc_ = ChildOf{this}

SubTree< ExprNode >	stridea_ = ChildOf{this}

SubTree< ExprNode >	strideb_ = ChildOf{this}

SubTree< ExprNode >	stridec_ = ChildOf{this}

SubTree< ExprNode >	batchSize_ = ChildOf{this}

bool	aIsRowMajor_

bool	bIsRowMajor_

bool	cIsRowMajor_

SubTree< StmtNode >	equivalent_

Additional Inherited Members
Public Types inherited from freetensor::EnableSelf< ASTPart >
typedef ASTPart	Self

Protected Member Functions inherited from freetensor::ASTPart
void	lock ()

void	unlock ()

virtual void	compHash ()=0

void	resetHash ()

Protected Attributes inherited from freetensor::ASTPart
size_t	hash_ = ~0ull

std::atomic_flag	lock_ = ATOMIC_FLAG_INIT

Detailed Description

External call to a batched GEMM

Member Function Documentation

◆ children()

std::vector< Stmt > freetensor::MatMulNode::children ( ) const

inlineoverridevirtual

Reimplemented from freetensor::StmtNode.

◆ compHash()

void freetensor::MatMulNode::compHash ( )

overridevirtual

Implements freetensor::ASTPart.

◆ DEFINE_NODE_TRAIT()

freetensor::MatMulNode::DEFINE_NODE_TRAIT ( MatMul )

Member Data Documentation

◆ a_

SubTree<ExprNode> freetensor::MatMulNode::a_ = ChildOf{this}

◆ aIsRowMajor_

bool freetensor::MatMulNode::aIsRowMajor_

◆ alpha_

SubTree<ExprNode> freetensor::MatMulNode::alpha_ = ChildOf{this}

◆ b_

SubTree<ExprNode> freetensor::MatMulNode::b_ = ChildOf{this}

◆ backend_

MatMulBackend freetensor::MatMulNode::backend_

◆ batchSize_

SubTree<ExprNode> freetensor::MatMulNode::batchSize_ = ChildOf{this}

◆ beta_

SubTree<ExprNode> freetensor::MatMulNode::beta_ = ChildOf{this}

◆ bIsRowMajor_

bool freetensor::MatMulNode::bIsRowMajor_

◆ c_

SubTree<ExprNode> freetensor::MatMulNode::c_ = ChildOf{this}

◆ cIsRowMajor_

bool freetensor::MatMulNode::cIsRowMajor_

◆ cutlassMicroKernelProperty_

SubTree<CutlassMicroKernelProperty, NullPolicy::Nullable> freetensor::MatMulNode::cutlassMicroKernelProperty_ = ChildOf{this}

◆ equivalent_

SubTree<StmtNode> freetensor::MatMulNode::equivalent_

Initial value:

= ChildOf{

this}

◆ k_

SubTree<ExprNode> freetensor::MatMulNode::k_ = ChildOf{this}

◆ lda_

SubTree<ExprNode> freetensor::MatMulNode::lda_ = ChildOf{this}

◆ ldb_

SubTree<ExprNode> freetensor::MatMulNode::ldb_ = ChildOf{this}

◆ ldc_

SubTree<ExprNode> freetensor::MatMulNode::ldc_ = ChildOf{this}

◆ m_

SubTree<ExprNode> freetensor::MatMulNode::m_ = ChildOf{this}

◆ n_

SubTree<ExprNode> freetensor::MatMulNode::n_ = ChildOf{this}

◆ stridea_

SubTree<ExprNode> freetensor::MatMulNode::stridea_ = ChildOf{this}

◆ strideb_

SubTree<ExprNode> freetensor::MatMulNode::strideb_ = ChildOf{this}

◆ stridec_

SubTree<ExprNode> freetensor::MatMulNode::stridec_ = ChildOf{this}

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

include/stmt.h
src/stmt.cc

Public Member Functions

Public Attributes

Additional Inherited Members