Struct MarkUnrollMutator

Defined in File lower_impl.h

Inheritance Relationships

Base Type

public ir::IRMutator< Expr *>

Struct Documentation

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

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

Public Functions

MarkUnrollMutator(const std::map<std::string, std::set<int>> &unrolls)

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>> unrolls

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