hoist_return_vars.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_HOIST_RETURN_VARS_H
#define FREE_TENSOR_HOIST_RETURN_VARS_H
 
#include <unordered_map>
#include <unordered_set>
 
#include <func.h>
#include <mutator.h>
 
namespace freetensor {
 
class HoistReturnVars : public Mutator {
    Func func_;
    ID outMostLoop_;
    std::vector<VarDef> toHoist_; // inner to outer
 
  public:
    HoistReturnVars(const Func &func) : func_(func) {}
 
  protected:
    Stmt visit(const VarDef &op) override;
    Stmt visit(const For &op) override;
};
 
Func hoistReturnVars(const Func &func);
 
} // namespace freetensor
 
#endif // FREE_TENSOR_HOIST_RETURN_VARS_H