MLIR  19.0.0git
Classes | Namespaces | Enumerations | Functions
LinalgInterfaces.cpp File Reference
#include "mlir/Dialect/Linalg/IR/LinalgInterfaces.h"
#include "mlir/Dialect/Affine/IR/AffineOps.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Arith/Utils/Utils.h"
#include "mlir/Dialect/Complex/IR/Complex.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/IR/AffineExprVisitor.h"
#include "mlir/IR/AffineMap.h"
#include "mlir/IR/TypeUtilities.h"
#include "llvm/ADT/SetOperations.h"
#include "llvm/ADT/SmallBitVector.h"
#include "llvm/ADT/SmallVector.h"
#include <algorithm>
#include "mlir/Dialect/Linalg/IR/LinalgInterfaces.cpp.inc"

Go to the source code of this file.

Classes

struct  HasAffineDimExprVisitor
 Visitor to check if any of the given set of positions from AffineDimExprs are used within an AffineExpr. More...
 

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::linalg
 
 mlir::linalg::detail
 

Enumerations

enum class  mlir::linalg::detail::MatchContractionResult {
  mlir::linalg::detail::Success = 0 , mlir::linalg::detail::NotLinalgOp , mlir::linalg::detail::WrongNumOperands , mlir::linalg::detail::NoReduction ,
  mlir::linalg::detail::NotProjectedPermutations , mlir::linalg::detail::NotAddMul
}
 
enum class  mlir::linalg::detail::MatchConvolutionResult {
  mlir::linalg::detail::Success = 0 , mlir::linalg::detail::NotLinalgOp , mlir::linalg::detail::WrongNumOperands , mlir::linalg::detail::WrongInputIndexingMap ,
  mlir::linalg::detail::NotProjectedPermutations , mlir::linalg::detail::NonConvolutionLoop , mlir::linalg::detail::OutputDimsNotParallel , mlir::linalg::detail::NonOutputDimNotReduction
}
 
enum class  MatchFillResult { Success = 0 , NotLinalgOp , WrongNumOperands , NotScalarInput }
 

Functions

static bool isaElemwiseSingleUnaryOrBinaryOpInterface (linalg::GenericOp genericOp, unsigned arity)
 
static Value getSourceSkipUnary (Value value)
 If the value is defined by a chain of unary side effect-free, go up the use-def chain until the first value that isn't defined by such an op. More...
 
template<typename AddOpTy , typename MulOpTy , typename... Args>
static bool isPairTemplateImpl (Operation *add, Operation *mul)
 Returns true if the two operations are of the kinds specified by a pair of consecutive template arguments. More...
 
template<typename... Args>
static bool isContractionBody (Block &block)
 Returns true if the block is a body of a contraction with the kinds of operations given pairwise by template arguments. More...
 
static llvm::SmallDenseSet< int64_t > findPermutationsIndexingOperand (AffineMap indexingMap, ArrayRef< utils::IteratorType > iterators, utils::IteratorType iter)
 Given an indexingMap and its corresponding iterators, returns the positions of the iterators of type iter that are indexed by the indexingMap as a permutation. More...
 
static FailureOr< SmallVector< utils::IteratorType > > inferIteratorsFromOutMap (AffineMap map)
 Infer the iterator types from the init affine map. More...
 
static FailureOr< ContractionDimensionsinferContractionDimsImpl (ArrayRef< AffineMap > indexingMaps, ArrayRef< utils::IteratorType > iterators)
 Find 2 parallel (m and n) and 1 reduction (k) dimension candidates that form a matmul subcomputation within linalgOp. More...
 
template<typename T >
static T getAffineExprOfType (AffineExpr lhs, AffineExpr rhs)
 Of the given two expressions returns one that is of type T (lhs gets preference over rhs) More...
 
static llvm::SmallDenseSet< int64_t > getPreservedDims (AffineMap map)
 
static SmallVector< int64_t, 2 > getConstantsFromExprList (const SmallVector< AffineExpr, 2 > &exprs)
 
static FailureOr< ConvolutionDimensionsinferConvolutionDimsImpl (LinalgOp linalgOp, ConvAccessExprWalker &inputExprWalker, bool allowEmptyConvolvedDims)
 Classifies dimensions in the linalgOp used by a convolution subcomputation, as captured by inputExprWalker. More...
 
static MatchFillResult isFillInterfaceImpl (Operation *op)
 
static std::pair< int64_t, int64_t > getResultsPositionInLoopsToShapeMap (LinalgOp &op)
 

Enumeration Type Documentation

◆ MatchFillResult

enum MatchFillResult
strong
Enumerator
Success 
NotLinalgOp 
WrongNumOperands 
NotScalarInput 

Definition at line 939 of file LinalgInterfaces.cpp.

Function Documentation

◆ findPermutationsIndexingOperand()

static llvm::SmallDenseSet<int64_t> findPermutationsIndexingOperand ( AffineMap  indexingMap,
ArrayRef< utils::IteratorType >  iterators,
utils::IteratorType  iter 
)
static

Given an indexingMap and its corresponding iterators, returns the positions of the iterators of type iter that are indexed by the indexingMap as a permutation.

This is useful to infer various subcomputations on a LinalgOp. This is performed by looking up each result in the indexingMap and determining whether:

  • It is a single AffineDimExpr.
  • It is the only result involving this AffineDimExpr.

Definition at line 280 of file LinalgInterfaces.cpp.

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

Referenced by inferContractionDimsImpl(), and inferConvolutionDimsImpl().

◆ getAffineExprOfType()

template<typename T >
static T getAffineExprOfType ( AffineExpr  lhs,
AffineExpr  rhs 
)
static

Of the given two expressions returns one that is of type T (lhs gets preference over rhs)

Definition at line 489 of file LinalgInterfaces.cpp.

◆ getConstantsFromExprList()

static SmallVector<int64_t, 2> getConstantsFromExprList ( const SmallVector< AffineExpr, 2 > &  exprs)
static

Definition at line 610 of file LinalgInterfaces.cpp.

Referenced by inferConvolutionDimsImpl().

◆ getPreservedDims()

static llvm::SmallDenseSet<int64_t> getPreservedDims ( AffineMap  map)
static

◆ getResultsPositionInLoopsToShapeMap()

static std::pair<int64_t, int64_t> getResultsPositionInLoopsToShapeMap ( LinalgOp &  op)
static

Definition at line 1048 of file LinalgInterfaces.cpp.

◆ getSourceSkipUnary()

static Value getSourceSkipUnary ( Value  value)
static

If the value is defined by a chain of unary side effect-free, go up the use-def chain until the first value that isn't defined by such an op.

Definition at line 174 of file LinalgInterfaces.cpp.

Referenced by mlir::linalg::detail::isContractionBody().

◆ inferContractionDimsImpl()

static FailureOr<ContractionDimensions> inferContractionDimsImpl ( ArrayRef< AffineMap indexingMaps,
ArrayRef< utils::IteratorType >  iterators 
)
static

Find 2 parallel (m and n) and 1 reduction (k) dimension candidates that form a matmul subcomputation within linalgOp.

These dimensions are such that:

  1. The m dimension is involved in an outer-product along LHS (i.e. it is a permutation on RES and LHS and does not appear in RHS).
  2. The n dimension is involved in an outer-product along RHS (i.e. it is a permutation on RES and RHS and does not appear in LHS).
  3. The k dimension appears as a permutation on LHS and RHS.
  4. m, n and k appear only once in any given indexing.
  5. Optional batch dimensions that appear in all operands are captured. This allows e.g. detecting that some contraction is embedded within linalgOp with some orthogonal heuristic.

Definition at line 329 of file LinalgInterfaces.cpp.

References findPermutationsIndexingOperand().

Referenced by mlir::linalg::inferContractionDims().

◆ inferConvolutionDimsImpl()

static FailureOr<ConvolutionDimensions> inferConvolutionDimsImpl ( LinalgOp  linalgOp,
ConvAccessExprWalker &  inputExprWalker,
bool  allowEmptyConvolvedDims 
)
static

Classifies dimensions in the linalgOp used by a convolution subcomputation, as captured by inputExprWalker.

If allowEmptyConvolvedDims is not set this this will fail if there is not at least convolved dimension pair (output image + filter loop). Convolution dimensions are specified in sorted order, and strides match the order of the filter loop dimensions, while the dilations match the order of the output image dimensions.

Definition at line 628 of file LinalgInterfaces.cpp.

References mlir::failure(), findPermutationsIndexingOperand(), and getConstantsFromExprList().

Referenced by mlir::linalg::inferConvolutionDims().

◆ inferIteratorsFromOutMap()

static FailureOr<SmallVector<utils::IteratorType> > inferIteratorsFromOutMap ( AffineMap  map)
static

Infer the iterator types from the init affine map.

This looks at which dims are present in the map results, and returns an iterator types array with parallel types for dims that are present, and reduction types for dims that are not present.

Definition at line 307 of file LinalgInterfaces.cpp.

References mlir::failure(), mlir::AffineMap::getNumDims(), mlir::AffineMap::getResults(), and mlir::AffineMap::isProjectedPermutation().

Referenced by mlir::linalg::inferContractionDims().

◆ isaElemwiseSingleUnaryOrBinaryOpInterface()

static bool isaElemwiseSingleUnaryOrBinaryOpInterface ( linalg::GenericOp  genericOp,
unsigned  arity 
)
static

Definition at line 106 of file LinalgInterfaces.cpp.

References mlir::Block::getOperations().

◆ isContractionBody()

template<typename... Args>
static bool isContractionBody ( Block block)
static

Returns true if the block is a body of a contraction with the kinds of operations given pairwise by template arguments.

Definition at line 268 of file LinalgInterfaces.cpp.

References mlir::linalg::detail::isContractionBody().

◆ isFillInterfaceImpl()

static MatchFillResult isFillInterfaceImpl ( Operation op)
static

Definition at line 946 of file LinalgInterfaces.cpp.

◆ isPairTemplateImpl()

template<typename AddOpTy , typename MulOpTy , typename... Args>
static bool isPairTemplateImpl ( Operation add,
Operation mul 
)
static

Returns true if the two operations are of the kinds specified by a pair of consecutive template arguments.

Definition at line 253 of file LinalgInterfaces.cpp.