analyze_linear.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_ANALYZE_LINEAR_H
#define FREE_TENSOR_ANALYZE_LINEAR_H
 
#include <optional>
#include <unordered_map>
#include <unordered_set>
 
#include <hash.h>
#include <math/linear.h>
#include <visitor.h>
 
namespace freetensor {
 
class AnalyzeLinear : public Visitor {
    std::unordered_map<AST, LinearExpr<int64_t>> result_;
 
  public:
    const std::unordered_map<AST, LinearExpr<int64_t>> &result() const {
        return result_;
    }
 
  protected:
    void visitExpr(const Expr &op) override;
 
    void visit(const IntConst &op) override;
    void visit(const Add &op) override;
    void visit(const Sub &op) override;
    void visit(const Mul &op) override;
    // Note that for integer (floored) div, k * a / d !== (k / d) * a, so we are
    // not handling Div here
};
 
LinearExpr<int64_t> linear(const Expr &expr);
 
std::optional<std::pair<LinearExpr<int64_t>, ASTNodeType>>
linearComp(const Expr &expr);
 
} // namespace freetensor
 
#endif // FREE_TENSOR_ANALYZE_LINEAR_H