Struct MarkVectorizeMutator

Defined in File lower_impl.h

Inheritance Relationships

Base Type

public ir::IRMutator< Expr *>

Struct Documentation

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

Mark the PolyFor as Vectorized if it is scheduled Vectorize in Stage.

Public Functions

MarkVectorizeMutator(const std::map<std::string, ir::VectorizeInfo> &vectorizes)

void operator()(Expr *expr)

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

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

Public Members

const std::map<std::string, ir::VectorizeInfo> &vectorizes

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