hoist_selected_var.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_HOIST_SELECTED_VAR_H
#define FREE_TENSOR_HOIST_SELECTED_VAR_H
 
#include <analyze/symbol_table.h>
#include <mutator.h>
#include <selector.h>
 
namespace freetensor {
 
class HoistSelectedVar : public SymbolTable<Mutator> {
    typedef SymbolTable<Mutator> BaseClass;
 
    const std::unordered_set<ID> &toHoist_;
 
  public:
    HoistSelectedVar(const std::unordered_set<ID> &toHoist)
        : toHoist_(toHoist) {}
 
  protected:
    using BaseClass::visit;
    // We use unsual modify-before-recursion in this Mutator, and therefore
    // there is no need to hoist a VarDef over another VarDef, and we can keep
    // relative position of VarDef nodes
    Stmt visit(const For &op) override;
    Stmt visit(const If &op) override;
    Stmt visit(const Assert &op) override;
    Stmt visit(const Assume &op) override;
    Stmt visit(const StmtSeq &op) override;
};
 
Stmt hoistSelectedVar(const Stmt &op, const std::string &selector);
Stmt hoistSelectedVar(const Stmt &op, const Ref<Selector> &selector);
} // namespace freetensor
 
#endif // FREE_TENSOR_HOIST_SELECTED_VAR_H