Struct MarkParallelMutator

Defined in File lower_impl.h

Inheritance Relationships

Base Type

public ir::IRMutator< Expr *>

Struct Documentation

struct cinn::lang::detail::MarkParallelMutator : public ir::IRMutator<Expr*>

Mark the PolyFor as Parallel if is called Parallel in Stage.

Public Functions

MarkParallelMutator(const std::map<std::string, std::set<int>> &parallels)

void operator()(Expr *expr)

void Visit(const ir::PolyFor *op, Expr *expr) override

void Visit(const ir::Store *op, Expr *expr) override

Public Members

std::map<std::string, std::set<int>> parallels

std::vector<ir::PolyFor*> stack