|
MLIR
22.0.0git
|
#include "mlir/Dialect/Vector/IR/VectorOps.h"#include "mlir/Conversion/ConvertToLLVM/ToLLVMInterface.h"#include "mlir/Dialect/Affine/IR/ValueBoundsOpInterfaceImpl.h"#include "mlir/Dialect/Arith/IR/Arith.h"#include "mlir/Dialect/Arith/Utils/Utils.h"#include "mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h"#include "mlir/Dialect/MemRef/IR/MemRef.h"#include "mlir/Dialect/Tensor/IR/Tensor.h"#include "mlir/Dialect/UB/IR/UBOps.h"#include "mlir/Dialect/Utils/IndexingUtils.h"#include "mlir/Dialect/Utils/StructuredOpsUtils.h"#include "mlir/IR/AffineExpr.h"#include "mlir/IR/AffineMap.h"#include "mlir/IR/Builders.h"#include "mlir/IR/BuiltinAttributes.h"#include "mlir/IR/BuiltinTypes.h"#include "mlir/IR/DialectImplementation.h"#include "mlir/IR/IRMapping.h"#include "mlir/IR/OpImplementation.h"#include "mlir/IR/PatternMatch.h"#include "mlir/IR/TypeUtilities.h"#include "mlir/IR/ValueRange.h"#include "mlir/Interfaces/SubsetOpInterface.h"#include "mlir/Interfaces/ValueBoundsOpInterface.h"#include "mlir/Support/LLVM.h"#include "mlir/Transforms/InliningUtils.h"#include "llvm/ADT/ArrayRef.h"#include "llvm/ADT/STLExtras.h"#include "llvm/ADT/SmallVector.h"#include "llvm/ADT/StringSet.h"#include "llvm/ADT/TypeSwitch.h"#include "llvm/Support/Casting.h"#include <cassert>#include <cstdint>#include "mlir/Dialect/Vector/IR/VectorDialect.cpp.inc"#include "mlir/Dialect/Vector/IR/VectorEnums.cpp.inc"#include "mlir/Dialect/Vector/IR/VectorAttributes.cpp.inc"#include "mlir/Dialect/Vector/IR/VectorOps.cpp.inc"Go to the source code of this file.
Classes | |
| struct | mlir::vector::detail::BitmaskEnumStorage |
| struct | CanonicalizeContractAdd< AddOpType > |
| Return a fused vector::ContractionOp which represents a patterns such as: More... | |
| class | FromElementsToShapeCast |
| Rewrite from_elements on multiple scalar extracts as a shape_cast on a single extract. More... | |
Namespaces | |
| mlir | |
| Include the generated interface declarations. | |
| mlir::vector | |
| mlir::vector::detail | |
Macros | |
| #define | GET_ATTRDEF_LIST |
| #define | GET_OP_LIST |
| #define | GET_ATTRDEF_CLASSES |
| #define | GET_OP_CLASSES |
Enumerations | |
| enum class | MaskFormat { AllTrue = 0 , AllFalse = 1 , Unknown = 2 } |
| Helper enum to classify mask value. More... | |
Functions | |
| static MaskFormat | getMaskFormat (Value mask) |
| Helper method to classify a mask value. More... | |
| static bool | isSupportedCombiningKind (CombiningKind combiningKind, Type elementType) |
| static unsigned | getEffectiveVectorRankForXferOp (ShapedType shapedType, VectorType vectorType) |
| Returns the effective rank of the vector to read/write for Xfer Ops. More... | |
| static bool | isSplatWriteConsistentWithMaskedRead (vector::TransferWriteOp write, vector::TransferReadOp read) |
Check if write is of a constant splat and the masked read is padded with the same splat value – meaning it could be the same value as the initial constant splat. More... | |
| static LogicalResult | incSlicePosition (MutableArrayRef< int64_t > position, ArrayRef< int64_t > shape, ArrayRef< int64_t > offsets) |
| static Attribute | convertNumericAttr (Attribute attr, Type expectedType) |
| Converts numeric attributes to the expected type. More... | |
| static bool | verifyDimMap (VectorType lhsType, VectorType rhsType, const std::vector< std::pair< int64_t, int64_t >> &map) |
| static LogicalResult | verifyOutputShape (ContractionOp op, VectorType lhsType, VectorType rhsType, Type accType, Type resType, const std::vector< std::pair< int64_t, int64_t >> &contractingDimMap, const std::vector< std::pair< int64_t, int64_t >> &batchDimMap) |
| static int64_t | getResultIndex (AffineMap map, AffineExpr targetExpr) |
| static std::vector< std::pair< int64_t, int64_t > > | getDimMap (ArrayRef< AffineMap > indexingMaps, ArrayAttr iteratorTypes, IteratorType targetIteratorType, MLIRContext *context) |
| static bool | isValidPositiveIndexOrPoison (int64_t index, int64_t poisonValue, int64_t maxIndex) |
| template<typename IntType > | |
| static SmallVector< IntType > | extractVector (ArrayAttr arrayAttr) |
| static LogicalResult | foldExtractOpFromExtractChain (ExtractOp extractOp) |
| Fold the result of chains of ExtractOp in place by simply concatenating the positions. More... | |
| static bool | hasZeroDimVectors (Operation *op) |
| Returns true if the operation has a 0-D vector type operand or result. More... | |
| static bool | isBroadcastLike (Operation *op) |
| All BroadcastOps and SplatOps, as well as ShapeCastOps that only prepend 1s, are considered to be 'broadcastlike'. More... | |
| static Value | foldExtractFromBroadcast (ExtractOp extractOp) |
| Fold extract(broadcast(X)) to either extract(X) or just X. More... | |
| static Value | foldExtractFromShuffle (ExtractOp extractOp) |
| Fold extractOp coming from ShuffleOp. More... | |
| static Value | foldExtractFromShapeCast (ExtractOp extractOp) |
| static Value | foldExtractFromExtractStrided (ExtractOp extractOp) |
| Fold an ExtractOp from ExtractStridedSliceOp. More... | |
| static Value | foldExtractStridedOpFromInsertChain (ExtractOp extractOp) |
| Fold extract_op fed from a chain of insertStridedSlice ops. More... | |
| static Value | foldScalarExtractFromFromElements (ExtractOp extractOp) |
| Try to fold the extraction of a scalar from a vector defined by vector.from_elements. More... | |
| template<typename OpType , typename AdaptorType > | |
| static Value | extractInsertFoldConstantOp (OpType op, AdaptorType adaptor, SmallVectorImpl< Value > &operands) |
If the dynamic indices of extractOp or insertOp are in fact constants, then fold it. More... | |
| static Attribute | foldPoisonIndexInsertExtractOp (MLIRContext *context, ArrayRef< int64_t > staticPos, int64_t poisonVal) |
| Fold an insert or extract operation into an poison value when a poison index is found at any dimension of the static position. More... | |
| static Attribute | foldPoisonSrcExtractOp (Attribute srcAttr) |
| Fold a vector extract from is a poison source. More... | |
| static Attribute | foldDenseElementsAttrSrcExtractOp (ExtractOp extractOp, Attribute srcAttr) |
| Fold a vector extract extracting from a DenseElementsAttr. More... | |
| static void | populateFromInt64AttrArray (ArrayAttr arrayAttr, SmallVectorImpl< int64_t > &results) |
| static bool | haveSameDefiningOp (OperandRange operands, Operation *defOp) |
Returns true if all the operands are defined by defOp. More... | |
| static LogicalResult | foldToElementsFromElements (ToElementsOp toElementsOp, SmallVectorImpl< OpFoldResult > &results) |
| Folds vector.to_elements(vector.from_elements(e0, e1, ...)) into (e0, e1, ...). More... | |
| static OpFoldResult | foldFromElementsToElements (FromElementsOp fromElementsOp) |
| Folds vector.from_elements(vector.to_elements(vector)) into vector. More... | |
| static OpFoldResult | foldFromElementsToConstant (FromElementsOp fromElementsOp, ArrayRef< Attribute > elements) |
| Fold vector.from_elements to a constant when all operands are constants. More... | |
| static LogicalResult | rewriteFromElementsAsBroadcast (FromElementsOp fromElementsOp, PatternRewriter &rewriter) |
| Rewrite vector.from_elements as vector.broadcast if the elements are the same. More... | |
| static llvm::SetVector< int64_t > | computeBroadcastedUnitDims (ArrayRef< int64_t > srcShape, ArrayRef< int64_t > dstShape) |
| Return the dimensions of the result vector that were formerly ones in the source tensor and thus correspond to "dim-1" broadcasting. More... | |
| static LogicalResult | foldBroadcastOfShapeCast (BroadcastOp broadcastOp) |
| template<typename T > | |
| static bool | isStepIndexArray (ArrayRef< T > idxArr, uint64_t begin, size_t width) |
| static int64_t | calculateInsertPosition (VectorType destTy, ArrayRef< int64_t > positions) |
| static Attribute | foldDenseElementsAttrDestInsertOp (InsertOp insertOp, Attribute srcAttr, Attribute dstAttr, int64_t maxVectorSizeFoldThreshold) |
| static Value | foldInsertUseChain (InsertOp insertOp) |
Folder to replace the dest operand of the insert op with the root dest of the insert op use chain. More... | |
| template<typename OpType > | |
| static LogicalResult | isIntegerArrayAttrSmallerThanShape (OpType op, ArrayAttr arrayAttr, ArrayRef< int64_t > shape, StringRef attrName) |
| template<typename OpType > | |
| static LogicalResult | isIntegerArrayAttrConfinedToRange (OpType op, ArrayAttr arrayAttr, int64_t min, int64_t max, StringRef attrName, bool halfOpen=true) |
| template<typename OpType > | |
| static LogicalResult | isIntegerArrayAttrConfinedToShape (OpType op, ArrayAttr arrayAttr, ArrayRef< int64_t > shape, StringRef attrName, bool halfOpen=true, int64_t min=0) |
| template<typename OpType > | |
| static LogicalResult | isSumOfIntegerArrayAttrConfinedToShape (OpType op, ArrayAttr arrayAttr1, ArrayAttr arrayAttr2, ArrayRef< int64_t > shape, StringRef attrName1, StringRef attrName2, bool halfOpen=true, int64_t min=1) |
| static ArrayAttr | makeI64ArrayAttr (ArrayRef< int64_t > values, MLIRContext *context) |
| static Type | inferStridedSliceOpResultType (VectorType vectorType, ArrayAttr offsets, ArrayAttr sizes, ArrayAttr strides) |
| static LogicalResult | foldExtractStridedOpFromInsertChain (ExtractStridedSliceOp op) |
| static OpFoldResult | foldExtractStridedSliceNonSplatConstant (ExtractStridedSliceOp op, Attribute foldInput) |
| template<typename EmitFun > | |
| static LogicalResult | verifyPermutationMap (AffineMap permutationMap, EmitFun emitOpError) |
| static LogicalResult | verifyTransferOp (VectorTransferOpInterface op, ShapedType shapedType, VectorType vectorType, VectorType maskType, VectorType inferredMaskType, AffineMap permutationMap, ArrayAttr inBounds) |
| static void | printTransferAttrs (OpAsmPrinter &p, VectorTransferOpInterface op) |
| template<typename TransferOp > | |
| static bool | isInBounds (TransferOp op, int64_t resultIdx, int64_t indicesIdx) |
| template<typename TransferOp > | |
| static LogicalResult | foldTransferInBoundsAttribute (TransferOp op) |
| template<typename TransferOp > | |
| static LogicalResult | foldTransferFullMask (TransferOp op) |
| #define GET_ATTRDEF_CLASSES |
| #define GET_ATTRDEF_LIST |
| #define GET_OP_CLASSES |
| #define GET_OP_LIST |
|
strong |
Helper enum to classify mask value.
| Enumerator | |
|---|---|
| AllTrue | |
| AllFalse | |
| Unknown | |
Definition at line 59 of file VectorOps.cpp.
|
static |
Definition at line 3314 of file VectorOps.cpp.
References computeStrides(), copy(), and mlir::linearize().
Referenced by foldDenseElementsAttrDestInsertOp().
|
static |
Return the dimensions of the result vector that were formerly ones in the source tensor and thus correspond to "dim-1" broadcasting.
Definition at line 2667 of file VectorOps.cpp.
Converts numeric attributes to the expected type.
Supports integer-to-integer and float-to-integer conversions. Returns the original attribute if no conversion is needed or supported.
Definition at line 402 of file VectorOps.cpp.
References mlir::get().
Referenced by foldDenseElementsAttrDestInsertOp(), and foldFromElementsToConstant().
|
static |
If the dynamic indices of extractOp or insertOp are in fact constants, then fold it.
Definition at line 2039 of file VectorOps.cpp.
References vectorShape().
|
static |
Definition at line 1388 of file VectorOps.cpp.
|
static |
Definition at line 2879 of file VectorOps.cpp.
References mlir::vector::isBroadcastableTo(), and min().
|
static |
Converts attribute to the expected type if there's a mismatch.
Definition at line 3489 of file VectorOps.cpp.
References calculateInsertPosition(), convertNumericAttr(), copy(), and mlir::get().
|
static |
Fold a vector extract extracting from a DenseElementsAttr.
Definition at line 2106 of file VectorOps.cpp.
References computeStrides(), copy(), mlir::get(), and mlir::linearize().
|
static |
Fold extract(broadcast(X)) to either extract(X) or just X.
Example:
broadcast extract [1][2]
(3, 4) -----—> (2, 3, 4) -------------—> (4)
becomes extract [1] (3,4) ----------------------------------—> (4)
The variable names used in this implementation correspond to the above shapes as,
input shape.broadcast shape.extract shape.This folding is possible when the suffix of input shape is the same as extract shape.
Definition at line 1709 of file VectorOps.cpp.
References mlir::decomposeMixedValues(), mlir::detail::enumerate(), mlir::Builder::getIndexAttr(), mlir::Operation::getOperand(), mlir::Value::getType(), and isBroadcastLike().
|
static |
Fold an ExtractOp from ExtractStridedSliceOp.
Definition at line 1871 of file VectorOps.cpp.
References hasZeroDimVectors().
|
static |
Definition at line 1806 of file VectorOps.cpp.
References delinearize(), and mlir::linearize().
|
static |
Fold extractOp coming from ShuffleOp.
Example:
shuffle = vector.shuffle a, b [0, 8, 7, 15] : vector<8xf32>, vector<8xf32> extract = vector.extract shuffle[3] : f32 from vector<4xf32> -> extract = vector.extract b[7] : f32 from vector<8xf32>
Definition at line 1774 of file VectorOps.cpp.
|
static |
Fold the result of chains of ExtractOp in place by simply concatenating the positions.
Definition at line 1396 of file VectorOps.cpp.
|
static |
Fold extract_op fed from a chain of insertStridedSlice ops.
Definition at line 1923 of file VectorOps.cpp.
References hasZeroDimVectors().
|
static |
Definition at line 4108 of file VectorOps.cpp.
References extractStrides(), and mlir::Builder::getI64ArrayAttr().
|
static |
Definition at line 4172 of file VectorOps.cpp.
References computeStrides(), copy(), mlir::get(), mlir::getI64SubArray(), incSlicePosition(), and mlir::linearize().
|
static |
Fold vector.from_elements to a constant when all operands are constants.
Example: c1 = arith.constant 1 : i32 c2 = arith.constant 2 : i32 v = vector.from_elements c1, c2 : vector<2xi32> => v = arith.constant dense<[1, 2]> : vector<2xi32>
Definition at line 2479 of file VectorOps.cpp.
References convertNumericAttr(), and mlir::get().
|
static |
Folds vector.from_elements(vector.to_elements(vector)) into vector.
Case #1: Input and output vectors are the same.
%0:3 = vector.to_elements a : vector<3xf32> %1 = vector.from_elements %0#0, %0#1, %0#2 : vector<3xf32> user_op %1
becomes:
user_op a
Definition at line 2446 of file VectorOps.cpp.
References mlir::Value::getType(), and haveSameDefiningOp().
|
static |
Folder to replace the dest operand of the insert op with the root dest of the insert op use chain.
Definition at line 3536 of file VectorOps.cpp.
References mlir::Value::getDefiningOp(), and mlir::Operation::setOperand().
|
static |
Fold an insert or extract operation into an poison value when a poison index is found at any dimension of the static position.
Definition at line 2088 of file VectorOps.cpp.
References mlir::get().
Fold a vector extract from is a poison source.
Definition at line 2098 of file VectorOps.cpp.
|
static |
Try to fold the extraction of a scalar from a vector defined by vector.from_elements.
E.g.:
%0 = vector.from_elements a, b : vector<2xf32> %1 = vector.extract %0[0] : f32 from vector<2xf32> ==> fold to a
Definition at line 2003 of file VectorOps.cpp.
References mlir::Value::getDefiningOp().
|
static |
Folds vector.to_elements(vector.from_elements(e0, e1, ...)) into (e0, e1, ...).
For example:
%0 = vector.from_elements a, b, c : vector<3xf32> %1:3 = vector.to_elements %0 : vector<3xf32> user_op %1#0, %1#1, %1#2
becomes:
user_op a, b, c
Definition at line 2404 of file VectorOps.cpp.
|
static |
Definition at line 4930 of file VectorOps.cpp.
References AllTrue, and getMaskFormat().
|
static |
Definition at line 4875 of file VectorOps.cpp.
References mlir::changed, mlir::Builder::getBoolArrayAttr(), mlir::AffineMap::getBroadcastDims(), mlir::AffineMap::getPermutationMap(), mlir::AffineMap::getResult(), and isInBounds().
|
static |
Definition at line 1144 of file VectorOps.cpp.
References mlir::detail::enumerate(), mlir::getAffineDimExpr(), and getResultIndex().
|
static |
Returns the effective rank of the vector to read/write for Xfer Ops.
When the element type of the shaped type is a scalar, this will simply return the rank of the vector ( the result for xfer_read or the value to store for xfer_write).
When the element type of the base shaped type is a vector, returns the difference between the original vector type and the element type of the shaped type.
EXAMPLE 1 (element type is a scalar):
EXAMPLE 2 (element type is a vector):
This is used to determine the number of minor dimensions for identity maps in vector transfer Ops.
Definition at line 178 of file VectorOps.cpp.
Referenced by mlir::vector::getTransferMinorIdentityMap().
|
static |
Helper method to classify a mask value.
Currently, the method looks "under the hood" of a constant value with dense attributes and a constant mask operation (since the client may be called at various stages during progressive lowering).
Definition at line 69 of file VectorOps.cpp.
References AllFalse, AllTrue, mlir::Value::getDefiningOp(), and Unknown.
Referenced by foldTransferFullMask().
|
static |
Definition at line 1136 of file VectorOps.cpp.
References mlir::AffineMap::getNumResults(), and mlir::AffineMap::getResult().
Referenced by getDimMap().
|
static |
Returns true if the operation has a 0-D vector type operand or result.
Definition at line 1657 of file VectorOps.cpp.
References mlir::Operation::getOperandTypes(), and mlir::Operation::getResultTypes().
Referenced by foldExtractFromExtractStrided(), and foldExtractStridedOpFromInsertChain().
|
static |
Returns true if all the operands are defined by defOp.
Otherwise, returns false.
Definition at line 2382 of file VectorOps.cpp.
References mlir::Value::getDefiningOp().
Referenced by foldFromElementsToElements().
|
static |
Definition at line 327 of file VectorOps.cpp.
Referenced by foldExtractStridedSliceNonSplatConstant().
|
static |
Definition at line 4016 of file VectorOps.cpp.
References mlir::get().
|
static |
All BroadcastOps and SplatOps, as well as ShapeCastOps that only prepend 1s, are considered to be 'broadcastlike'.
Definition at line 1669 of file VectorOps.cpp.
Referenced by foldExtractFromBroadcast().
|
static |
Definition at line 4858 of file VectorOps.cpp.
References getConstantIntValue().
Referenced by foldTransferInBoundsAttribute().
|
static |
Definition at line 3618 of file VectorOps.cpp.
|
static |
Definition at line 3638 of file VectorOps.cpp.
References mlir::detail::enumerate(), max(), and min().
|
static |
Definition at line 3603 of file VectorOps.cpp.
|
static |
Check if write is of a constant splat and the masked read is padded with the same splat value – meaning it could be the same value as the initial constant splat.
Definition at line 206 of file VectorOps.cpp.
References mlir::DenseElementsAttr::getSplatValue(), mlir::DenseElementsAttr::isSplat(), mlir::m_Constant(), and mlir::matchPattern().
|
static |
Definition at line 3025 of file VectorOps.cpp.
|
static |
Definition at line 3661 of file VectorOps.cpp.
References mlir::detail::enumerate(), max(), and min().
|
static |
Definition at line 131 of file VectorOps.cpp.
References mlir::Type::isIntOrIndex(), mlir::Type::isIntOrIndexOrFloat(), and MINUI.
|
static |
Definition at line 1278 of file VectorOps.cpp.
|
static |
Definition at line 3682 of file VectorOps.cpp.
References mlir::get().
|
static |
Definition at line 2362 of file VectorOps.cpp.
|
static |
Definition at line 4684 of file VectorOps.cpp.
References mlir::OpAsmPrinter::printOptionalAttrDict().
|
static |
Rewrite vector.from_elements as vector.broadcast if the elements are the same.
Example: %0 = vector.from_elements a, a, a : vector<3xf32> => %0 = vector.broadcast a : f32 to vector<3xf32>
Definition at line 2517 of file VectorOps.cpp.
References mlir::RewriterBase::replaceOpWithNewOp().
|
static |
Definition at line 925 of file VectorOps.cpp.
|
static |
Definition at line 936 of file VectorOps.cpp.
References mlir::AffineMap::compose(), mlir::AffineMap::get(), mlir::get(), mlir::getAffineConstantExpr(), mlir::AffineMap::getNumInputs(), mlir::AffineMap::getResults(), mlir::getUnusedDimsBitVector(), and mlir::simplifyAffineMap().
|
static |
Definition at line 4576 of file VectorOps.cpp.
References mlir::AffineMap::getNumInputs(), and mlir::AffineMap::getResults().
|
static |
Definition at line 4605 of file VectorOps.cpp.
References mlir::get(), mlir::AffineMap::getNumInputs(), mlir::AffineMap::getNumResults(), mlir::AffineMap::getNumSymbols(), and mlir::DataLayout::getTypeSizeInBits().
1.9.1