#include <as_matmul.h>

Inheritance diagram for freetensor::AsMatMul:

Collaboration diagram for freetensor::AsMatMul:

Public Member Functions
	AsMatMul (const ID &loop, MatMulBackend backend)

bool	done () const

const ID &	resultId () const

const auto &	dimsABatch () const

const auto &	dimsBBatch () const

const auto &	dimsCBatch () const

const auto &	dimsAM () const

const auto &	dimsAK () const

const auto &	dimsBK () const

const auto &	dimsBN () const

const auto &	dimsCM () const

const auto &	dimsCN () const

const auto &	defIdA () const

const auto &	defIdB () const

const auto &	defIdC () const

Public Member Functions inherited from freetensor::SymbolTable< Mutator >
	SymbolTable (T &&...args)

const std::unordered_set< std::string > &	names () const override

const std::unordered_map< std::string, VarDef > &	defs () const override

const std::unordered_map< std::string, For > &	loops () const override

bool	hasDef (const std::string &name) const override

const VarDef &	def (const std::string &name) const override

Ref< Buffer >	buffer (const std::string &name) const override

bool	hasLoop (const std::string &name) const override

const For &	loop (const std::string &name) const override

void	pushDef (const VarDef &op) override

void	popDef (const VarDef &op) override

void	pushFor (const For &op) override

void	popFor (const For &op) override

const SymbolTableData &	symbolTableSnapshot () const

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

virtual Stmt	operator() (const Stmt &op) final

virtual Expr	operator() (const Expr &op) final

virtual const std::unordered_set< std::string > &	names () const =0

virtual const std::unordered_map< std::string, VarDef > &	defs () const =0

virtual const std::unordered_map< std::string, For > &	loops () const =0

virtual bool	hasDef (const std::string &name) const =0

virtual const VarDef &	def (const std::string &name) const =0

virtual Ref< Buffer >	buffer (const std::string &name) const =0

virtual bool	hasLoop (const std::string &name) const =0

virtual const For &	loop (const std::string &name) const =0

virtual void	pushDef (const VarDef &op)=0

virtual void	popDef (const VarDef &op)=0

virtual void	pushFor (const For &op)=0

virtual void	popFor (const For &op)=0

Protected Member Functions
Stmt	visitStmt (const Stmt &op) override

Stmt	visit (const For &op) override

Stmt	visit (const ReduceTo &op) override

Stmt	visit (const Store &op) override

Stmt	visit (const VarDef &op) override

Protected Member Functions inherited from freetensor::SymbolTable< Mutator >
BaseClass::StmtRetType	visit (const VarDef &op) override

BaseClass::StmtRetType	visit (const For &op) override

Protected Member Functions inherited from freetensor::Mutator
virtual Expr	visitExpr (const Expr &op)

virtual Stmt	visitStmt (const Stmt &op)

virtual Stmt	visit (const Any &op)

virtual Stmt	visit (const StmtSeq &op)

virtual Stmt	visit (const VarDef &op)

virtual Expr	visit (const Var &op)

virtual Stmt	visit (const Store &op)

virtual Stmt	visit (const Alloc &op)

virtual Stmt	visit (const Free &op)

virtual Expr	visit (const Load &op)

virtual Stmt	visit (const ReduceTo &op)

virtual Expr	visit (const AnyExpr &op)

virtual Expr	visit (const IntConst &op)

virtual Expr	visit (const FloatConst &op)

virtual Expr	visit (const BoolConst &op)

virtual Expr	visit (const Add &op)

virtual Expr	visit (const Sub &op)

virtual Expr	visit (const Mul &op)

virtual Expr	visit (const RealDiv &op)

virtual Expr	visit (const FloorDiv &op)

virtual Expr	visit (const CeilDiv &op)

virtual Expr	visit (const RoundTowards0Div &op)

virtual Expr	visit (const Mod &op)

virtual Expr	visit (const Remainder &op)

virtual Expr	visit (const Min &op)

virtual Expr	visit (const Max &op)

virtual Expr	visit (const LT &op)

virtual Expr	visit (const LE &op)

virtual Expr	visit (const GT &op)

virtual Expr	visit (const GE &op)

virtual Expr	visit (const EQ &op)

virtual Expr	visit (const NE &op)

virtual Expr	visit (const LAnd &op)

virtual Expr	visit (const LOr &op)

virtual Expr	visit (const LNot &op)

virtual Expr	visit (const Sqrt &op)

virtual Expr	visit (const Exp &op)

virtual Expr	visit (const Ln &op)

virtual Expr	visit (const Square &op)

virtual Expr	visit (const Sigmoid &op)

virtual Expr	visit (const Sin &op)

virtual Expr	visit (const Cos &op)

virtual Expr	visit (const Tan &op)

virtual Expr	visit (const Tanh &op)

virtual Expr	visit (const Abs &op)

virtual Expr	visit (const Floor &op)

virtual Expr	visit (const Ceil &op)

virtual Expr	visit (const Unbound &op)

virtual Stmt	visit (const For &op)

virtual Stmt	visit (const If &op)

virtual Stmt	visit (const Assert &op)

virtual Stmt	visit (const Assume &op)

virtual Expr	visit (const IfExpr &op)

virtual Expr	visit (const Cast &op)

virtual Expr	visit (const Intrinsic &op)

virtual Stmt	visit (const Eval &op)

virtual Stmt	visit (const MatMul &op)

virtual Stmt	visit (const MarkVersion &op)

virtual Expr	visit (const LoadAtVersion &op)

Additional Inherited Members
Public Types inherited from freetensor::Mutator
typedef Expr	ExprRetType

typedef Stmt	StmtRetType

Constructor & Destructor Documentation

◆ AsMatMul()

freetensor::AsMatMul::AsMatMul	(	const ID &	loop,
		MatMulBackend	backend
	)

inline

Member Function Documentation

◆ defIdA()

const auto & freetensor::AsMatMul::defIdA ( ) const

inline

◆ defIdB()

const auto & freetensor::AsMatMul::defIdB ( ) const

inline

◆ defIdC()

const auto & freetensor::AsMatMul::defIdC ( ) const

inline

◆ dimsABatch()

const auto & freetensor::AsMatMul::dimsABatch ( ) const

inline

◆ dimsAK()

const auto & freetensor::AsMatMul::dimsAK ( ) const

inline

◆ dimsAM()

const auto & freetensor::AsMatMul::dimsAM ( ) const

inline

◆ dimsBBatch()

const auto & freetensor::AsMatMul::dimsBBatch ( ) const

inline

◆ dimsBK()

const auto & freetensor::AsMatMul::dimsBK ( ) const

inline

◆ dimsBN()

const auto & freetensor::AsMatMul::dimsBN ( ) const

inline

◆ dimsCBatch()

const auto & freetensor::AsMatMul::dimsCBatch ( ) const

inline

◆ dimsCM()

const auto & freetensor::AsMatMul::dimsCM ( ) const

inline

◆ dimsCN()

const auto & freetensor::AsMatMul::dimsCN ( ) const

inline

◆ done()

bool freetensor::AsMatMul::done ( ) const

inline

◆ resultId()

const ID & freetensor::AsMatMul::resultId ( ) const

inline

◆ visit() [1/4]

Stmt freetensor::AsMatMul::visit ( const For & op )

overrideprotectedvirtual

Reimplemented from freetensor::Mutator.

◆ visit() [2/4]

Stmt freetensor::AsMatMul::visit ( const ReduceTo & op )

overrideprotectedvirtual

Reimplemented from freetensor::Mutator.

◆ visit() [3/4]

Stmt freetensor::AsMatMul::visit ( const Store & op )

overrideprotectedvirtual

Reimplemented from freetensor::Mutator.

◆ visit() [4/4]

Stmt freetensor::AsMatMul::visit ( const VarDef & op )

overrideprotectedvirtual

Reimplemented from freetensor::Mutator.

◆ visitStmt()

Stmt freetensor::AsMatMul::visitStmt ( const Stmt & op )

overrideprotectedvirtual

Reimplemented from freetensor::Mutator.

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

include/schedule/as_matmul.h
src/schedule/as_matmul.cc

Public Member Functions

Protected Member Functions

Additional Inherited Members

Constructor & Destructor Documentation

◆ AsMatMul()

Member Function Documentation

◆ defIdA()

◆ defIdB()

◆ defIdC()

◆ dimsABatch()

◆ dimsAK()

◆ dimsAM()

◆ dimsBBatch()

◆ dimsBK()

◆ dimsBN()

◆ dimsCBatch()

◆ dimsCM()

◆ dimsCN()

◆ done()

◆ resultId()

◆ visit() [1/4]

◆ visit() [2/4]

◆ visit() [3/4]

◆ visit() [4/4]

◆ visitStmt()