merge_and_hoist_if.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_MERGE_AND_HOIST_IF_H
#define FREE_TENSOR_MERGE_AND_HOIST_IF_H
 
#include <unordered_set>
 
#include <analyze/symbol_table.h>
#include <func.h>
#include <mutator.h>
 
namespace freetensor {
 
class MergeAndHoistIf : public SymbolTable<Mutator> {
    typedef SymbolTable<Mutator> BaseClass;
 
    bool isFixPoint_ = true;
 
  public:
    bool isFixPoint() const { return isFixPoint_; }
 
  protected:
    Stmt visit(const StmtSeq &op) override;
    Stmt visit(const VarDef &op) override;
    Stmt visit(const For &op) override;
};
 
Stmt mergeAndHoistIf(const Stmt &op);
 
DEFINE_PASS_FOR_FUNC(mergeAndHoistIf)
 
} // namespace freetensor
 
#endif // FREE_TENSOR_MERGE_AND_HOIST_IF_H