check_loop_order.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_CHECK_LOOP_ORDER_H
#define FREE_TENSOR_CHECK_LOOP_ORDER_H
 
#include <string>
#include <unordered_set>
#include <vector>
 
#include <visitor.h>
 
namespace freetensor {
 
class CheckLoopOrder : public Visitor {
    std::vector<ID> dstOrder_;
    std::vector<For> curOrder_, outerLoops_, outerLoopStack_;
    std::vector<StmtSeq> stmtSeqStack_, stmtSeqInBetween_;
    std::unordered_set<std::string> itersDefinedInNest_;
    bool reqireRangeDefinedOutside_;
    bool done_ = false;
 
  public:
    CheckLoopOrder(const std::vector<ID> &dstOrder,
                   bool reqireRangeDefinedOutside = true)
        : dstOrder_(dstOrder),
          reqireRangeDefinedOutside_(reqireRangeDefinedOutside) {}
 
    const std::vector<For> &order() const;
 
    const std::vector<StmtSeq> &stmtSeqInBetween() const {
        return stmtSeqInBetween_;
    }
 
    const std::vector<For> &outerLoops() const { return outerLoops_; }
 
  protected:
    void visitStmt(const Stmt &stmt) override;
    void visit(const For &op) override;
    void visit(const StmtSeq &op) override;
};
 
} // namespace freetensor
 
#endif // FREE_TENSOR_CHECK_LOOP_ORDER_H