FreeTensor
Loading...
Searching...
No Matches
Public Member Functions | Protected Member Functions | List of all members
freetensor::CheckLoopOrder Class Reference

#include <check_loop_order.h>

Inheritance diagram for freetensor::CheckLoopOrder:
Inheritance graph
[legend]
Collaboration diagram for freetensor::CheckLoopOrder:
Collaboration graph
[legend]

Public Member Functions

 CheckLoopOrder (const std::vector< ID > &dstOrder, bool reqireRangeDefinedOutside=true)
 
const std::vector< For > & order () const
 
const std::vector< StmtSeq > & stmtSeqInBetween () const
 
const std::vector< For > & outerLoops () const
 
- Public Member Functions inherited from freetensor::Visitor
virtual ~Visitor ()
 
virtual void operator() (const AST &op) final
 

Protected Member Functions

void visitStmt (const Stmt &stmt) override
 
void visit (const For &op) override
 
void visit (const StmtSeq &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
 

Detailed Description

Return loops in nesting order

Parameters
requireRangeDefinedOutside: If true, throw InvalidSchedule when the range of any loop is an expression of the iterating variable of other loops in the nest. For example, for i = 0 to 4 { for j = 0 to i {}} will be illegal.

Constructor & Destructor Documentation

◆ CheckLoopOrder()

freetensor::CheckLoopOrder::CheckLoopOrder ( const std::vector< ID > &  dstOrder,
bool  reqireRangeDefinedOutside = true 
)
inline

Member Function Documentation

◆ order()

const std::vector< For > & freetensor::CheckLoopOrder::order ( ) const

All required loops, sorted from outer to inner

◆ outerLoops()

const std::vector< For > & freetensor::CheckLoopOrder::outerLoops ( ) const
inline

Loops surrounding all loops in dstOrder

◆ stmtSeqInBetween()

const std::vector< StmtSeq > & freetensor::CheckLoopOrder::stmtSeqInBetween ( ) const
inline

All StmtSeq nodes nested inside the outer-most required loop, and nesting the inner-most required loop, sorted from outer to inner

◆ visit() [1/2]

void freetensor::CheckLoopOrder::visit ( const For op)
overrideprotectedvirtual

Reimplemented from freetensor::Visitor.

◆ visit() [2/2]

void freetensor::CheckLoopOrder::visit ( const StmtSeq op)
overrideprotectedvirtual

Reimplemented from freetensor::Visitor.

◆ visitStmt()

void freetensor::CheckLoopOrder::visitStmt ( const Stmt stmt)
overrideprotectedvirtual

Reimplemented from freetensor::Visitor.


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