normalize_var_in_kernel.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_GPU_NORMALIZE_VAR_IN_KERNEL_H
#define FREE_TENSOR_GPU_NORMALIZE_VAR_IN_KERNEL_H
 
#include <unordered_map>
#include <unordered_set>
 
#include <analyze/comp_transient_bounds.h>
#include <analyze/symbol_table.h>
#include <func.h>
#include <memory>
#include <mutator.h>
 
namespace freetensor {
 
namespace gpu {
 
class NormalizeVarInKernel : public CompTransientBounds<SymbolTable<Mutator>> {
    typedef CompTransientBounds<SymbolTable<Mutator>> BaseClass;
 
    std::unordered_set<std::string> legalNames_;
 
    std::vector<VarDef> varsToHoist_;
    std::unordered_set<std::string> usedNamesInKernel_;
    std::unordered_map<std::string, int> nameCntInKernel_;
    bool inKernel_ = false;
 
  protected:
    using BaseClass::visit;
    Stmt visit(const VarDef &op) override;
    Stmt visit(const For &op) override;
};
 
Stmt normalizeVarInKernel(const Stmt &s);
 
DEFINE_PASS_FOR_FUNC(normalizeVarInKernel)
 
} // namespace gpu
 
} // namespace freetensor
 
#endif // FREE_TENSOR_GPU_NORMALIZE_VAR_IN_KERNEL_H