hoist_var_over_stmt_seq.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_HOIST_VAR_OVER_STMT_SEQ_H
#define FREE_TENSOR_HOIST_VAR_OVER_STMT_SEQ_H
 
#include <optional>
#include <unordered_map>
 
#include <func.h>
#include <mutator.h>
#include <pass/rename_var.h>
 
namespace freetensor {
 
class HoistVarOverStmtSeq : public RenameVar {
    typedef RenameVar BaseClass;
 
    std::optional<std::vector<ID>> togetherIds_;
    bool isFixPoint_ = true;
 
  public:
    HoistVarOverStmtSeq(
        const std::optional<std::vector<ID>> &togetherIds = std::nullopt)
        : togetherIds_(togetherIds) {}
 
    bool isFixPoint() const { return isFixPoint_; }
 
  protected:
    using BaseClass::visit;
    Stmt visit(const StmtSeq &op) override;
};
 
Stmt hoistVarOverStmtSeq(
    const Stmt &op,
    const std::optional<std::vector<ID>> &togetherIds = std::nullopt);
 
DEFINE_PASS_FOR_FUNC(hoistVarOverStmtSeq)
 
} // namespace freetensor
 
#endif // FREE_TENSOR_HOIST_VAR_OVER_STMT_SEQ_H