MLIR  16.0.0git
Classes | Namespaces | Enumerations | Functions
VectorRewritePatterns.h File Reference
#include <utility>
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/Dialect/Vector/Utils/VectorUtils.h"
#include "mlir/IR/BuiltinOps.h"
#include "mlir/IR/PatternMatch.h"
+ Include dependency graph for VectorRewritePatterns.h:
+ This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  mlir::vector::VectorTransformsOptions
 Structure to control the behavior of vector transform patterns. More...
 
struct  mlir::vector::UnrollVectorOptions
 Options that control the vector unrolling. More...
 
struct  mlir::vector::VectorTransferFullPartialRewriter
 Apply splitFullAndPartialTransfer selectively via a pattern. More...
 
class  mlir::vector::ContractionOpToMatmulOpLowering
 Progressive lowering of a vector.contract a, b, c with row-major matmul semantics to: More...
 
class  mlir::vector::ContractionOpToOuterProductOpLowering
 Progressive lowering of a vector.contract a, b, c with row-major matmul semantics to a reduction_size-unrolled sequence: More...
 
class  mlir::vector::ContractionOpToDotLowering
 Progressive lowering of a vector.contract a, b, c with row-major matmul semantics to an output-size-unrolled sequence: More...
 
class  mlir::vector::ContractionOpLowering
 Progressive lowering of ContractionOp. More...
 

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::vector
 

Enumerations

enum class  mlir::vector::VectorTransposeLowering { mlir::vector::EltWise = 0 , mlir::vector::Flat = 1 , mlir::vector::Shuffle = 2 }
 Enum to control the lowering of vector.transpose operations. More...
 
enum class  mlir::vector::VectorMultiReductionLowering { mlir::vector::InnerParallel = 0 , mlir::vector::InnerReduction = 1 }
 Enum to control the lowering of vector.multi_reduction operations. More...
 
enum class  mlir::vector::VectorContractLowering { mlir::vector::Dot = 0 , mlir::vector::Matmul = 1 , mlir::vector::OuterProduct = 2 , mlir::vector::ParallelArith = 3 }
 Enum to control the lowering of vector.contract operations. More...
 
enum class  mlir::vector::VectorTransferSplit { mlir::vector::None = 0 , mlir::vector::VectorTransfer = 1 , mlir::vector::LinalgCopy = 2 , mlir::vector::ForceInBounds = 3 }
 Enum to control the splitting of vector.transfer operations into in-bounds and out-of-bounds variants. More...
 

Functions

void mlir::vector::populateVectorTransposeLoweringPatterns (RewritePatternSet &patterns, VectorTransformsOptions options=VectorTransformsOptions(), PatternBenefit benefit=1)
 Insert TransposeLowering patterns into extraction/insertion. More...
 
void mlir::vector::populateVectorMultiReductionLoweringPatterns (RewritePatternSet &patterns, VectorMultiReductionLowering options, PatternBenefit benefit=1)
 Collect a set of patterns to convert vector.multi_reduction op into a sequence of vector.reduction ops. More...
 
void mlir::vector::populateVectorContractLoweringPatterns (RewritePatternSet &patterns, VectorTransformsOptions options=VectorTransformsOptions(), PatternBenefit benefit=1)
 Collects patterns to progressively lower vector contraction ops on high-D into low-D reduction and product ops. More...
 
void mlir::vector::populateVectorReductionToContractPatterns (RewritePatternSet &patterns, PatternBenefit benefit=1)
 Collect patterns to convert reduction op to vector.contract and fold transpose/broadcast ops into the contract. More...
 
void mlir::vector::populateVectorScanLoweringPatterns (RewritePatternSet &patterns, PatternBenefit benefit=1)
 Collect patterns to convert scan op. More...
 
void mlir::vector::populateVectorTransferPermutationMapLoweringPatterns (RewritePatternSet &patterns, PatternBenefit benefit=1)
 Collect a set of transfer read/write lowering patterns that simplify the permutation map (e.g., converting it to a minor identity map) by inserting broadcasts and transposes. More...
 
void mlir::vector::populateVectorTransferCollapseInnerMostContiguousDimsPatterns (RewritePatternSet &patterns, PatternBenefit benefit=1)
 Collect a set of patterns to reduce the rank of the operands of vector transfer ops to operate on the largest contigious vector. More...
 
void mlir::vector::populateVectorInsertExtractStridedSliceDecompositionPatterns (RewritePatternSet &patterns, PatternBenefit benefit=1)
 Populate patterns with the following patterns. More...
 
void mlir::vector::populateVectorExtractStridedSliceToExtractInsertChainPatterns (RewritePatternSet &patterns, std::function< bool(ExtractStridedSliceOp)> controlFn=nullptr, PatternBenefit benefit=1)
 Populate patterns with a pattern to breaks down 1-D extract_strided_slice ops into a chain of Extract ops to extract each element from the source, and then a chain of Insert ops to insert to the target vector. More...
 
void mlir::vector::populateVectorInsertExtractStridedSliceTransforms (RewritePatternSet &patterns, PatternBenefit benefit=1)
 Populate patterns with the following patterns. More...
 
void mlir::vector::populateVectorUnrollPatterns (RewritePatternSet &patterns, const UnrollVectorOptions &options, PatternBenefit benefit=1)
 Collect a set of pattern to unroll vector operations to a smaller shapes. More...