MLIR  19.0.0git
Functions
AffineMap.cpp File Reference
#include "mlir/IR/AffineMap.h"
#include "AffineMapDetail.h"
#include "mlir/IR/AffineExpr.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/Support/LogicalResult.h"
#include "mlir/Support/MathExtras.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/SmallBitVector.h"
#include "llvm/ADT/SmallSet.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/raw_ostream.h"
#include <iterator>
#include <numeric>
#include <optional>
#include <type_traits>

Go to the source code of this file.

Functions

template<typename AffineExprContainer >
static SmallVector< AffineMap, 4 > inferFromExprList (ArrayRef< AffineExprContainer > exprsList, MLIRContext *context)
 Creates an affine map each for each list of AffineExpr's in exprsList while inferring the right number of dimensional and symbolic inputs needed based on the maximum dimensional and symbolic identifier appearing in the expressions. More...
 
static SmallVector< AffineMapcompressUnusedListImpl (ArrayRef< AffineMap > maps, llvm::function_ref< AffineMap(AffineMap)> compressionFun)
 Implementation detail to compress multiple affine maps with a compressionFun that is expected to be either compressUnusedDims or compressUnusedSymbols. More...
 
template<typename AffineDimOrSymExpr >
static AffineMap projectCommonImpl (AffineMap map, const llvm::SmallBitVector &toProject, bool compress)
 Common implementation to project out dimensions or symbols from an affine map based on the template type. More...
 

Function Documentation

◆ compressUnusedListImpl()

static SmallVector<AffineMap> compressUnusedListImpl ( ArrayRef< AffineMap maps,
llvm::function_ref< AffineMap(AffineMap)>  compressionFun 
)
static

Implementation detail to compress multiple affine maps with a compressionFun that is expected to be either compressUnusedDims or compressUnusedSymbols.

The implementation keeps track of num dims and symbols across the different affine maps.

Definition at line 650 of file AffineMap.cpp.

References mlir::AffineMap::get(), mlir::AffineMap::getNumDims(), mlir::AffineMap::getNumSymbols(), and mlir::AffineMap::getResults().

Referenced by mlir::compressUnusedDims(), and mlir::compressUnusedSymbols().

◆ inferFromExprList()

template<typename AffineExprContainer >
static SmallVector<AffineMap, 4> inferFromExprList ( ArrayRef< AffineExprContainer >  exprsList,
MLIRContext context 
)
static

Creates an affine map each for each list of AffineExpr's in exprsList while inferring the right number of dimensional and symbolic inputs needed based on the maximum dimensional and symbolic identifier appearing in the expressions.

Definition at line 281 of file AffineMap.cpp.

References mlir::AffineMap::get(), and mlir::getMaxDimAndSymbol().

Referenced by mlir::AffineMap::inferFromExprList().

◆ projectCommonImpl()

template<typename AffineDimOrSymExpr >
static AffineMap projectCommonImpl ( AffineMap  map,
const llvm::SmallBitVector &  toProject,
bool  compress 
)
static

Common implementation to project out dimensions or symbols from an affine map based on the template type.

Additionally, if 'compress' is true, the projected out dimensions or symbols are also dropped from the resulting map.

Definition at line 822 of file AffineMap.cpp.

References mlir::AffineMap::get(), mlir::getAffineConstantExpr(), mlir::getAffineDimExpr(), mlir::getAffineSymbolExpr(), mlir::AffineMap::getContext(), mlir::AffineMap::getNumDims(), mlir::AffineMap::getNumResults(), mlir::AffineMap::getNumSymbols(), mlir::AffineMap::getResults(), mlir::AffineExpr::replaceDims(), and mlir::AffineExpr::replaceSymbols().