check_not_modified.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_CHECK_NOT_MODIFIED
#define FREE_TENSOR_CHECK_NOT_MODIFIED
 
#include <optional>
#include <unordered_map>
#include <unordered_set>
 
#include <analyze/symbol_table.h>
#include <mutator.h>
#include <visitor.h>
 
namespace freetensor {
 
enum class CheckNotModifiedSide : int { Before, After };
 
class CheckNameToDefMapping : public SymbolTable<Visitor> {
    typedef SymbolTable<Visitor> BaseClass;
 
    ID pos_;
    const std::unordered_set<std::string> &names_;
    std::unordered_map<std::string, ID> name2def_;
 
  public:
    CheckNameToDefMapping(const ID &pos,
                          const std::unordered_set<std::string> &names)
        : pos_(pos), names_(names) {}
 
    const std::unordered_map<std::string, ID> &name2def() const {
        return name2def_;
    }
 
  protected:
    void visitStmt(const Stmt &stmt) override;
};
 
class CheckReadToParallelScopeMapping : public SymbolTable<Visitor> {
    typedef SymbolTable<Visitor> BaseClass;
 
    ID pos_;
    const std::unordered_set<std::string> &reads_;
    std::unordered_map<std::string, ID> read2scope_;
 
  public:
    CheckReadToParallelScopeMapping(
        const ID &pos, const std::unordered_set<std::string> &reads)
        : pos_(pos), reads_(reads) {}
 
    const auto &read2scope() const { return read2scope_; }
 
  protected:
    void visitStmt(const Stmt &stmt) override;
};
 
class InsertTmpEval : public Mutator {
    Expr s0Expr_, s1Expr_;
    ID s0_, s1_, s0Eval_, s1Eval_;
    CheckNotModifiedSide s0Side_, s1Side_;
 
  public:
    InsertTmpEval(const Expr &s0Expr, const Expr &s1Expr,
                  CheckNotModifiedSide s0Side, const ID &s0,
                  CheckNotModifiedSide s1Side, const ID &s1)
        : s0Expr_(s0Expr), s1Expr_(s1Expr), s0_(s0), s1_(s1), s0Side_(s0Side),
          s1Side_(s1Side) {}
 
    const ID &s0Eval() const { return s0Eval_; }
    const ID &s1Eval() const { return s1Eval_; }
 
  protected:
    Stmt visitStmt(const Stmt &op) override;
};
 
std::optional<bool>
checkNotModifiedFastPreCheck(const Stmt &op, const Expr &expr,
                             CheckNotModifiedSide s0Side, const ID &s0,
                             CheckNotModifiedSide s1Side, const ID &s1);
std::optional<bool>
checkNotModifiedFastPreCheck(const Stmt &op, const Expr &s0Expr,
                             const Expr &s1Expr, CheckNotModifiedSide s0Side,
                             const ID &s0, CheckNotModifiedSide s1Side,
                             const ID &s1);
bool checkNotModified(const Stmt &op, const Expr &expr,
                      CheckNotModifiedSide s0Side, const ID &s0,
                      CheckNotModifiedSide s1Side, const ID &s1);
 
bool checkNotModified(const Stmt &op, const Expr &s0Expr, const Expr &s1Expr,
                      CheckNotModifiedSide s0Side, const ID &s0,
                      CheckNotModifiedSide s1Side, const ID &s1);
 
} // namespace freetensor
 
#endif // FREE_TENSOR_CHECK_NOT_MODIFIED