#include <analyze_version.h>

Inheritance diagram for freetensor::AnalyzeVersion:

Collaboration diagram for freetensor::AnalyzeVersion:

Public Member Functions
	AnalyzeVersion (const ID &def, const std::unordered_set< ID > &affectingScopes, const std::unordered_set< ID > &needVersions, const std::unordered_set< ID > &fakeStoreIds, const std::unordered_map< Stmt, Expr > &scopeLen, const Expr &totLen, std::unordered_map< StmtOrExprID, Expr > &versions, std::unordered_map< std::string, std::pair< std::string, Expr > > &userVersions)

const std::string &	tapeName () const

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

virtual void	operator() (const AST &op) final

Protected Member Functions
void	visit (const Load &op) override

void	visit (const MarkVersion &op) override

void	visit (const Store &op) override

void	visit (const ReduceTo &op) override

void	visit (const VarDef &op) override

void	visit (const For &op) override

void	visit (const StmtSeq &op) override

Protected Member Functions inherited from freetensor::TrackStmt< Visitor >
void	pushStmt (const Stmt &op)

void	popStmt (const Stmt &)

const Stmt &	curStmt () const

BaseClass::StmtRetType	visitStmt (const Stmt &op) override

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

virtual void	visitStmt (const Stmt &op)

virtual void	visit (const Any &op)

virtual void	visit (const AnyExpr &op)

virtual void	visit (const Func &op)

virtual void	visit (const StmtSeq &op)

virtual void	visit (const VarDef &op)

virtual void	visit (const Var &op)

virtual void	visit (const Store &op)

virtual void	visit (const Alloc &op)

virtual void	visit (const Free &op)

virtual void	visit (const Load &op)

virtual void	visit (const ReduceTo &op)

virtual void	visit (const IntConst &op)

virtual void	visit (const FloatConst &op)

virtual void	visit (const BoolConst &op)

virtual void	visit (const Add &op)

virtual void	visit (const Sub &op)

virtual void	visit (const Mul &op)

virtual void	visit (const RealDiv &op)

virtual void	visit (const FloorDiv &op)

virtual void	visit (const CeilDiv &op)

virtual void	visit (const RoundTowards0Div &op)

virtual void	visit (const Mod &op)

virtual void	visit (const Remainder &op)

virtual void	visit (const Min &op)

virtual void	visit (const Max &op)

virtual void	visit (const LT &op)

virtual void	visit (const LE &op)

virtual void	visit (const GT &op)

virtual void	visit (const GE &op)

virtual void	visit (const EQ &op)

virtual void	visit (const NE &op)

virtual void	visit (const LAnd &op)

virtual void	visit (const LOr &op)

virtual void	visit (const LNot &op)

virtual void	visit (const Sqrt &op)

virtual void	visit (const Exp &op)

virtual void	visit (const Ln &op)

virtual void	visit (const Square &op)

virtual void	visit (const Sigmoid &op)

virtual void	visit (const Sin &op)

virtual void	visit (const Cos &op)

virtual void	visit (const Tan &op)

virtual void	visit (const Tanh &op)

virtual void	visit (const Abs &op)

virtual void	visit (const Floor &op)

virtual void	visit (const Ceil &op)

virtual void	visit (const Unbound &op)

virtual void	visit (const For &op)

virtual void	visit (const If &op)

virtual void	visit (const Assert &op)

virtual void	visit (const Assume &op)

virtual void	visit (const IfExpr &op)

virtual void	visit (const Cast &op)

virtual void	visit (const Intrinsic &op)

virtual void	visit (const Eval &op)

virtual void	visit (const MatMul &op)

virtual void	visit (const MarkVersion &op)

virtual void	visit (const LoadAtVersion &op)

Additional Inherited Members
Public Types inherited from freetensor::Visitor
typedef void	ExprRetType

typedef void	StmtRetType

Constructor & Destructor Documentation

◆ AnalyzeVersion()

freetensor::AnalyzeVersion::AnalyzeVersion	(	const ID &	def,
		const std::unordered_set< ID > &	affectingScopes,
		const std::unordered_set< ID > &	needVersions,
		const std::unordered_set< ID > &	fakeStoreIds,
		const std::unordered_map< Stmt, Expr > &	scopeLen,
		const Expr &	totLen,
		std::unordered_map< StmtOrExprID, Expr > &	versions,
		std::unordered_map< std::string, std::pair< std::string, Expr > > &	userVersions
	)

inline

Member Function Documentation

◆ tapeName()

const std::string & freetensor::AnalyzeVersion::tapeName ( ) const

inline

◆ visit() [1/7]

void freetensor::AnalyzeVersion::visit ( const For & op )

overrideprotectedvirtual

Reimplemented from freetensor::Visitor.

◆ visit() [2/7]

void freetensor::AnalyzeVersion::visit ( const Load & op )

overrideprotectedvirtual

Reimplemented from freetensor::Visitor.

◆ visit() [3/7]

void freetensor::AnalyzeVersion::visit ( const MarkVersion & op )

overrideprotectedvirtual

Reimplemented from freetensor::Visitor.

◆ visit() [4/7]

void freetensor::AnalyzeVersion::visit ( const ReduceTo & op )

overrideprotectedvirtual

Reimplemented from freetensor::Visitor.

◆ visit() [5/7]

void freetensor::AnalyzeVersion::visit ( const StmtSeq & op )

overrideprotectedvirtual

Reimplemented from freetensor::Visitor.

◆ visit() [6/7]

void freetensor::AnalyzeVersion::visit ( const Store & op )

overrideprotectedvirtual

Reimplemented from freetensor::Visitor.

◆ visit() [7/7]

void freetensor::AnalyzeVersion::visit ( const VarDef & op )

overrideprotectedvirtual

Reimplemented from freetensor::Visitor.

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

include/autograd/analyze_version.h
src/autograd/analyze_version.cc

Public Member Functions

Protected Member Functions

Additional Inherited Members

Constructor & Destructor Documentation

◆ AnalyzeVersion()

Member Function Documentation

◆ tapeName()

◆ visit() [1/7]

◆ visit() [2/7]

◆ visit() [3/7]

◆ visit() [4/7]

◆ visit() [5/7]

◆ visit() [6/7]

◆ visit() [7/7]