mlir::x86vector Namespace Reference

Namespaces
namespace	avx2
	Helpers extracted from:

Classes
struct	MaskHelper
	Helper class to factor out the creation and extraction of masks from nibs. More...

Functions
void	registerTransformDialectExtension (DialectRegistry &registry)
void	populateVectorContractToFMAPatterns (RewritePatternSet &patterns)
void	populateVectorContractToPackedTypeDotProductPatterns (RewritePatternSet &patterns)
void	populateVectorContractBF16ToFMAPatterns (RewritePatternSet &patterns)
void	populateSinkVectorProducerOpsPatterns (RewritePatternSet &patterns)
bool	isInVnniLayout (Operation *op, llvm::ArrayRef< AffineMap > indexingMaps, std::optional< unsigned > blockingFactor=std::nullopt)
static FailureOr< SmallVector< mlir::utils::IteratorType > >	inferIteratorsFromOutMap (AffineMap map)

Function Documentation

inferIteratorsFromOutMap()

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

static

Definition at line 22 of file X86VectorUtils.cpp.

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

Referenced by isInVnniLayout().

isInVnniLayout()

bool mlir::x86vector::isInVnniLayout	(	Operation *	op,
		llvm::ArrayRef< AffineMap >	indexingMaps,
		std::optional< unsigned >	blockingFactor = std::nullopt )

Definition at line 35 of file X86VectorUtils.cpp.

References mlir::Operation::getOperand(), mlir::AffineMap::getResult(), mlir::linalg::inferContractionDims(), and inferIteratorsFromOutMap().

Referenced by VectorContractBF16ToFMA::matchAndRewrite().

populateSinkVectorProducerOpsPatterns()

void mlir::x86vector::populateSinkVectorProducerOpsPatterns

(

RewritePatternSet &

patterns

)

Definition at line 144 of file SinkVectorProducerOps.cpp.

References mlir::patterns.

populateVectorContractBF16ToFMAPatterns()

void mlir::x86vector::populateVectorContractBF16ToFMAPatterns

(

RewritePatternSet &

patterns

)

Definition at line 324 of file VectorContractBF16ToFMA.cpp.

References mlir::patterns.

populateVectorContractToFMAPatterns()

void mlir::x86vector::populateVectorContractToFMAPatterns

(

RewritePatternSet &

patterns

)

Definition at line 140 of file VectorContractToFMA.cpp.

References mlir::patterns.

populateVectorContractToPackedTypeDotProductPatterns()

void mlir::x86vector::populateVectorContractToPackedTypeDotProductPatterns

(

RewritePatternSet &

patterns

)

Definition at line 213 of file VectorContractToPackedTypeDotProduct.cpp.

References mlir::patterns.

registerTransformDialectExtension()

void mlir::x86vector::registerTransformDialectExtension

(

DialectRegistry &

registry

)

Definition at line 71 of file X86VectorTransformOps.cpp.

References mlir::DialectRegistry::addExtensions().

Referenced by mlir::registerAllExtensions().