#include "mlir/Dialect/Vector/Transforms/IndexedAccessOpInterfaceImpl.h"
#include "mlir/Dialect/MemRef/IR/MemoryAccessOpInterfaces.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/IR/Dialect.h"
#include "mlir/IR/IRMapping.h"
#include "mlir/IR/Operation.h"
#include "mlir/IR/PatternMatch.h"
#include <type_traits>

Functions
template<typename LoadStoreOp>
static bool	hasMemrefSemantics (Operation *op)
	Return true if this op has the memref semantics expected by this model.
template<typename LoadStoreOp>
static constexpr bool	supportsRankZeroVectorAccess ()
	Return true if this op supports rank-0 vector operands/results.
static unsigned	getNumLeadingUnitDims (VectorType vecTy)
	Return the number of leading static unit dimensions in vecTy.
static SmallVector< int64_t >	getAccessedVectorShape (VectorType vecTy, bool supportsRankZero)
	Return the vector shape whose access strides must be preserved, omitting redundant leading static unit dimensions and marking scalable dimensions as dynamic.
static VectorType	dropLeadingDims (VectorType vecTy, unsigned numLeadingDims)
	Return vecTy with numLeadingDims dimensions dropped from the front.
static std::optional< VectorType >	getShapeCastTypeForOperand (Value operand, VectorType vecTy, unsigned numLeadingDims)
	Return the shape-cast type for vector operands that match vecTy.

Function Documentation

◆ dropLeadingDims()

VectorType dropLeadingDims	(	VectorType	vecTy,
		unsigned	numLeadingDims )

static

Return vecTy with numLeadingDims dimensions dropped from the front.

Definition at line 77 of file IndexedAccessOpInterfaceImpl.cpp.

Referenced by getShapeCastTypeForOperand().

◆ getAccessedVectorShape()

SmallVector< int64_t > getAccessedVectorShape	(	VectorType	vecTy,
		bool	supportsRankZero )

static

Return the vector shape whose access strides must be preserved, omitting redundant leading static unit dimensions and marking scalable dimensions as dynamic.

If the op cannot access rank-0 vectors, preserve one trailing unit dimension instead of returning an empty shape.

Definition at line 61 of file IndexedAccessOpInterfaceImpl.cpp.

References getNumLeadingUnitDims().