move_out_first_or_last_iter.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_MOVE_OUT_FIRST_OR_LAST_ITER_H
#define FREE_TENSOR_MOVE_OUT_FIRST_OR_LAST_ITER_H
 
#include <unordered_map>
 
#include <func.h>
#include <pass/z3_simplify.h>
 
namespace freetensor {
 
class MoveOutFirstOrLastIter : public Z3SimplifyWithSymbolTable {
  protected:
    Stmt visit(const For &op) override;
};
 
inline Stmt moveOutFirstOrLastIter(const Stmt &op) {
    return MoveOutFirstOrLastIter()(op);
}
 
DEFINE_PASS_FOR_FUNC(moveOutFirstOrLastIter)
 
} // namespace freetensor
 
#endif // FREE_TENSOR_MOVE_OUT_FIRST_OR_LAST_ITER_H