#include "mlir/IR/AffineExprVisitor.h"

Inheritance diagram for mlir::SimpleAffineExprFlattener:

Public Member Functions
	SimpleAffineExprFlattener (unsigned numDims, unsigned numSymbols)

virtual	~SimpleAffineExprFlattener ()=default

LogicalResult	visitMulExpr (AffineBinaryOpExpr expr)

LogicalResult	visitAddExpr (AffineBinaryOpExpr expr)

LogicalResult	visitDimExpr (AffineDimExpr expr)

LogicalResult	visitSymbolExpr (AffineSymbolExpr expr)

LogicalResult	visitConstantExpr (AffineConstantExpr expr)

LogicalResult	visitCeilDivExpr (AffineBinaryOpExpr expr)

LogicalResult	visitFloorDivExpr (AffineBinaryOpExpr expr)

LogicalResult	visitModExpr (AffineBinaryOpExpr expr)

Public Member Functions inherited from mlir::AffineExprVisitor< SimpleAffineExprFlattener, LogicalResult >
LogicalResult	walkPostOrder (AffineExpr expr)

Public Member Functions inherited from mlir::AffineExprVisitorBase< SubClass, RetTy >
RetTy	visit (AffineExpr expr)

RetTy	visitAffineBinaryOpExpr (AffineBinaryOpExpr expr)

RetTy	visitAddExpr (AffineBinaryOpExpr expr)

RetTy	visitMulExpr (AffineBinaryOpExpr expr)

RetTy	visitModExpr (AffineBinaryOpExpr expr)

RetTy	visitFloorDivExpr (AffineBinaryOpExpr expr)

RetTy	visitCeilDivExpr (AffineBinaryOpExpr expr)

RetTy	visitConstantExpr (AffineConstantExpr expr)

RetTy	visitDimExpr (AffineDimExpr expr)

RetTy	visitSymbolExpr (AffineSymbolExpr expr)

Public Attributes
std::vector< SmallVector< int64_t, 8 > >	operandExprStack

unsigned	numDims

unsigned	numSymbols

unsigned	numLocals

SmallVector< AffineExpr, 4 >	localExprs

Protected Member Functions
virtual void	addLocalFloorDivId (ArrayRef< int64_t > dividend, int64_t divisor, AffineExpr localExpr)

virtual LogicalResult	addLocalIdSemiAffine (ArrayRef< int64_t > lhs, ArrayRef< int64_t > rhs, AffineExpr localExpr)
	Add a local identifier (needed to flatten a mod, floordiv, ceildiv, mul expr) when the rhs is a symbolic expression. More...

Detailed Description

Definition at line 354 of file AffineExprVisitor.h.

Constructor & Destructor Documentation

◆ SimpleAffineExprFlattener()

SimpleAffineExprFlattener::SimpleAffineExprFlattener	(	unsigned	numDims,
		unsigned	numSymbols
	)

Definition at line 1263 of file AffineExpr.cpp.

References operandExprStack.

◆ ~SimpleAffineExprFlattener()

virtual mlir::SimpleAffineExprFlattener::~SimpleAffineExprFlattener ( )

virtualdefault

Member Function Documentation

◆ addLocalFloorDivId()

void SimpleAffineExprFlattener::addLocalFloorDivId	(	ArrayRef< int64_t >	dividend,
		int64_t	divisor,
		AffineExpr	localExpr
	)

protectedvirtual

Definition at line 1537 of file AffineExpr.cpp.

References localExprs, numLocals, and operandExprStack.

Referenced by visitModExpr().

◆ addLocalIdSemiAffine()

LogicalResult SimpleAffineExprFlattener::addLocalIdSemiAffine	(	ArrayRef< int64_t >	lhs,
		ArrayRef< int64_t >	rhs,
		AffineExpr	localExpr
	)

protectedvirtual

Add a local identifier (needed to flatten a mod, floordiv, ceildiv, mul expr) when the rhs is a symbolic expression.

The local identifier added may be a floordiv, ceildiv, mul or mod of a pure affine/semi-affine function of other identifiers, coefficients of which are specified in the lhs of the mod, floordiv, ceildiv or mul expression and with respect to a symbolic rhs expression. localExpr is the simplified tree expression (AffineExpr) corresponding to the quantifier.

Definition at line 1548 of file AffineExpr.cpp.

References localExprs, numLocals, and operandExprStack.