MLIR  19.0.0git
Public Member Functions | Static Public Member Functions | List of all members
mlir::affine::AffineValueMap Class Reference

An AffineValueMap is an affine map plus its ML value operands and results for analysis purposes. More...

#include "mlir/Dialect/Affine/IR/AffineValueMap.h"

Public Member Functions

 AffineValueMap ()=default
 
 AffineValueMap (AffineMap map, ValueRange operands, ValueRange results={})
 
 ~AffineValueMap ()
 
void reset (AffineMap map, ValueRange operands, ValueRange results={})
 
void composeSimplifyAndCanonicalize ()
 Composes all incoming affine.apply ops and then simplifies and canonicalizes the map and operands. More...
 
bool isMultipleOf (unsigned idx, int64_t factor) const
 Return true if the idx^th result can be proved to be a multiple of 'factor', false otherwise. More...
 
bool isFunctionOf (unsigned idx, Value value) const
 Return true if the idx^th result depends on 'value', false otherwise. More...
 
bool isConstant (unsigned idx) const
 Return true if the result at 'idx' is a constant, false otherwise. More...
 
bool isIdentity () const
 Return true if this is an identity map. More...
 
void setResult (unsigned i, AffineExpr e)
 
AffineExpr getResult (unsigned i)
 
unsigned getNumOperands () const
 
unsigned getNumDims () const
 
unsigned getNumSymbols () const
 
unsigned getNumResults () const
 
Value getOperand (unsigned i) const
 
ArrayRef< ValuegetOperands () const
 
AffineMap getAffineMap () const
 
LogicalResult canonicalize ()
 Attempts to canonicalize the map and operands. More...
 

Static Public Member Functions

static void difference (const AffineValueMap &a, const AffineValueMap &b, AffineValueMap *res)
 Return the value map that is the difference of value maps 'a' and 'b', represented as an affine map and its operands. More...
 

Detailed Description

An AffineValueMap is an affine map plus its ML value operands and results for analysis purposes.

The structure is still a tree form that is same as that of an affine map or an AffineApplyOp. However, its operands, results, and its map can themselves change as a result of substitutions, simplifications, and other analysis.

Definition at line 35 of file AffineValueMap.h.

Constructor & Destructor Documentation

◆ AffineValueMap() [1/2]

mlir::affine::AffineValueMap::AffineValueMap ( )
default

◆ AffineValueMap() [2/2]

AffineValueMap::AffineValueMap ( AffineMap  map,
ValueRange  operands,
ValueRange  results = {} 
)

Definition at line 15 of file AffineValueMap.cpp.

◆ ~AffineValueMap()

AffineValueMap::~AffineValueMap ( )
default

Member Function Documentation

◆ canonicalize()

LogicalResult AffineValueMap::canonicalize ( )

Attempts to canonicalize the map and operands.

Return success if the map and/or operands have been modified.

Definition at line 3947 of file AffineOps.cpp.

References composeAffineMapAndOperands(), mlir::failure(), and mlir::success().

◆ composeSimplifyAndCanonicalize()

void AffineValueMap::composeSimplifyAndCanonicalize ( )

Composes all incoming affine.apply ops and then simplifies and canonicalizes the map and operands.

This can change the number of operands, but the result count remains the same.

Definition at line 27 of file AffineValueMap.cpp.

References mlir::affine::canonicalizeMapAndOperands(), mlir::affine::fullyComposeAffineMapAndOperands(), getAffineMap(), and mlir::MutableAffineMap::reset().

Referenced by isAccessIndexInvariant(), and mlir::affine::isInvariantAccess().

◆ difference()

void AffineValueMap::difference ( const AffineValueMap a,
const AffineValueMap b,
AffineValueMap res 
)
static

Return the value map that is the difference of value maps 'a' and 'b', represented as an affine map and its operands.

The output map + operands are canonicalized and simplified.

Definition at line 36 of file AffineValueMap.cpp.

References mlir::affine::canonicalizeMapAndOperands(), mlir::affine::fullyComposeAffineMapAndOperands(), mlir::AffineMap::get(), getAffineMap(), getNumDims(), getNumOperands(), getNumResults(), getNumSymbols(), getOperands(), reset(), mlir::AffineMap::shiftDims(), mlir::AffineMap::shiftSymbols(), and mlir::simplifyAffineMap().

Referenced by mlir::affine::getTripCountMapAndOperands(), and mlir::affine::MemRefAccess::operator==().

◆ getAffineMap()

AffineMap AffineValueMap::getAffineMap ( ) const

◆ getNumDims()

unsigned mlir::affine::AffineValueMap::getNumDims ( ) const
inline

Definition at line 75 of file AffineValueMap.h.

References mlir::AffineMap::getNumDims().

Referenced by difference().

◆ getNumOperands()

unsigned mlir::affine::AffineValueMap::getNumOperands ( ) const
inline

Definition at line 74 of file AffineValueMap.h.

Referenced by difference().

◆ getNumResults()

unsigned mlir::affine::AffineValueMap::getNumResults ( ) const
inline

◆ getNumSymbols()

unsigned mlir::affine::AffineValueMap::getNumSymbols ( ) const
inline

Definition at line 76 of file AffineValueMap.h.

References mlir::AffineMap::getNumSymbols().

Referenced by difference().

◆ getOperand()

Value AffineValueMap::getOperand ( unsigned  i) const

Definition at line 102 of file AffineValueMap.cpp.

Referenced by mlir::affine::getRelationFromMap().

◆ getOperands()

ArrayRef< Value > AffineValueMap::getOperands ( ) const

◆ getResult()

AffineExpr mlir::affine::AffineValueMap::getResult ( unsigned  i)
inline

Definition at line 73 of file AffineValueMap.h.

References mlir::AffineMap::getResult().

Referenced by mlir::affine::getTripCountMapAndOperands().

◆ isConstant()

bool mlir::affine::AffineValueMap::isConstant ( unsigned  idx) const

Return true if the result at 'idx' is a constant, false otherwise.

◆ isFunctionOf()

bool AffineValueMap::isFunctionOf ( unsigned  idx,
Value  value 
) const

Return true if the idx^th result depends on 'value', false otherwise.

This method uses the invariant that operands are always positionally aligned with the AffineDimExpr in the underlying AffineMap.

Definition at line 91 of file AffineValueMap.cpp.

References findIndex(), getAffineMap(), mlir::AffineMap::getResult(), and mlir::AffineExpr::isFunctionOfDim().

Referenced by isAccessIndexInvariant().

◆ isIdentity()

bool mlir::affine::AffineValueMap::isIdentity ( ) const

Return true if this is an identity map.

◆ isMultipleOf()

bool AffineValueMap::isMultipleOf ( unsigned  idx,
int64_t  factor 
) const
inline

Return true if the idx^th result can be proved to be a multiple of 'factor', false otherwise.

Definition at line 85 of file AffineValueMap.cpp.

References mlir::MutableAffineMap::isMultipleOf().

◆ reset()

void AffineValueMap::reset ( AffineMap  map,
ValueRange  operands,
ValueRange  results = {} 
)

◆ setResult()

void mlir::affine::AffineValueMap::setResult ( unsigned  i,
AffineExpr  e 
)
inline

Definition at line 72 of file AffineValueMap.h.

Referenced by mlir::affine::getTripCountMapAndOperands().


The documentation for this class was generated from the following files: