min_max.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_MIN_MAX_H
#define FREE_TENSOR_MIN_MAX_H
 
#include <functional>
 
#include <expr.h>
 
namespace freetensor {
 
Expr makeMinMaxImpl(const std::vector<std::vector<Expr>> &exprs,
                    const std::function<Expr()> &inf,
                    const std::function<Expr()> &negInf);
Expr makeMaxMinImpl(const std::vector<std::vector<Expr>> &exprs,
                    const std::function<Expr()> &inf,
                    const std::function<Expr()> &negInf);
 
inline std::function<Expr()> asExprGenerator(std::nullptr_t) {
    return []() { return nullptr; };
}
inline std::function<Expr()> asExprGenerator(const std::function<Expr()> &e) {
    return e;
}
inline std::function<Expr()> asExprGenerator(const Expr &e) {
    return [&e]() { return e; };
}
 
template <typename T, typename U>
Expr makeMinMax(const std::vector<std::vector<Expr>> &exprs, const T &inf,
                const U &negInf) {
    return makeMinMaxImpl(exprs, asExprGenerator(inf), asExprGenerator(negInf));
}
 
template <typename T, typename U>
Expr makeMaxMin(const std::vector<std::vector<Expr>> &exprs, const T &negInf,
                const U &inf) {
    return makeMaxMinImpl(exprs, asExprGenerator(negInf), asExprGenerator(inf));
}
 
Expr makeLOrLAnd(const std::vector<std::vector<Expr>> &exprs);
 
} // namespace freetensor
 
#endif // FREE_TENSOR_MIN_MAX_H