inlining.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_INLINING_H
#define FREE_TENSOR_INLINING_H
 
#include <unordered_map>
 
#include <analyze/track_stmt.h>
#include <mutator.h>
 
namespace freetensor {
 
class MakeInline : public TrackStmt<Mutator> {
    typedef TrackStmt<Mutator> BaseClass;
 
    ID def_;
    std::string var_;
    const std::unordered_map<Load, Expr> &replace_;
 
  public:
    MakeInline(const ID &def, const std::unordered_map<Load, Expr> &replace)
        : def_(def), replace_(replace) {}
 
  protected:
    using BaseClass::visit;
    Expr visit(const Load &op) override;
    Stmt visit(const Store &op) override;
    Stmt visit(const ReduceTo &op) override;
    Stmt visit(const VarDef &op) override;
};
 
Stmt inlining(const Stmt &ast, const ID &def);
 
} // namespace freetensor
 
#endif // FREE_TENSOR_INLINING_H