use_builtin_div.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_USE_BUILTIN_DIV_H
#define FREE_TENSOR_USE_BUILTIN_DIV_H
 
#include <analyze/comp_transient_bounds.h>
#include <analyze/comp_unique_bounds.h>
#include <analyze/symbol_table.h>
#include <func.h>
#include <memory>
#include <mutator.h>
 
namespace freetensor {
 
class UseBuiltinDiv : public CompTransientBounds<SymbolTable<Mutator>> {
    typedef CompTransientBounds<SymbolTable<Mutator>> BaseClass;
 
    Ref<CompUniqueBounds> bound_;
 
  protected:
    using BaseClass::visit;
    Stmt visitStmt(const Stmt &op) override;
    Expr visit(const FloorDiv &op) override;
    Expr visit(const CeilDiv &op) override;
    Expr visit(const Mod &op) override;
};
 
Stmt useBuiltinDiv(const Stmt &op);
 
DEFINE_PASS_FOR_FUNC(useBuiltinDiv)
 
} // namespace freetensor
 
#endif // FREE_TENSOR_USE_BUILTIN_DIV_H