lower_parallel_reduction.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_CPU_LOWER_PARALLEL_REDUCTION_H
#define FREE_TENSOR_CPU_LOWER_PARALLEL_REDUCTION_H
 
#include <unordered_map>
#include <unordered_set>
 
#include <analyze/symbol_table.h>
#include <func.h>
#include <mutator.h>
 
namespace freetensor {
 
namespace cpu {
 
class LowerParallelReduction : public SymbolTable<Mutator> {
    typedef SymbolTable<Mutator> BaseClass;
 
    std::vector<For> loopStack_;
    std::unordered_set<std::string> needRedirect_;
 
  private:
    std::vector<std::pair<For, int>> reducedBy(const ReduceTo &op);
 
  protected:
    using BaseClass::visit;
    Stmt visit(const For &op) override;
    Stmt visit(const ReduceTo &op) override;
};
 
Stmt lowerParallelReduction(const Stmt &op);
 
DEFINE_PASS_FOR_FUNC(lowerParallelReduction)
 
} // namespace cpu
 
} // namespace freetensor
 
#endif // FREE_TENSOR_CPU_LOWER_PARALLEL_REDUCTION_H