annotate_conds.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_ANNOTATE_CONDS_H
#define FREE_TENSOR_ANNOTATE_CONDS_H
 
#include <mutator.h>
 
namespace freetensor {
 
class AnnotateConds : public Mutator {
    std::vector<Expr> conds_; // stack of conditions, may be null
 
  private:
    void addCond(const Expr &expr);
 
  protected:
    Stmt visit(const StmtSeq &op) override;
    Stmt visit(const For &op) override;
    Stmt visit(const If &op) override;
    Stmt visit(const Assert &op) override;
    Stmt visit(const Assume &op) override;
};
 
inline Stmt annotateConds(const Stmt &op) { return AnnotateConds()(op); }
 
} // namespace freetensor
 
#endif // FREE_TENSOR_ANNOTATE_CONDS_H