Lower vector.contract with all size one reduction dimensions to elementwise ops when possible. More...

Inheritance diagram for ContractOpToElementwise:

Public Types
using	FilterConstraintType
Public Types inherited from mlir::OpRewritePattern< vector::ContractionOp >
using	Base
	Type alias to allow derived classes to inherit constructors with using Base::Base;.

Public Member Functions
	ContractOpToElementwise (vector::VectorContractLowering vectorContractLowering, MLIRContext *context, PatternBenefit benefit=1, const FilterConstraintType &constraint=defaultFilter)
FailureOr< Value >	matchAndRewriteMaskableOp (vector::ContractionOp contractOp, MaskingOpInterface maskOp, PatternRewriter &rewriter) const override
Public Member Functions inherited from mlir::OpRewritePattern< vector::ContractionOp >
	OpRewritePattern (MLIRContext *context, PatternBenefit benefit=1, ArrayRef< StringRef > generatedNames={})
	Patterns must specify the root operation name they match against, and can also specify the benefit of the pattern matching and a list of generated ops.
Public Member Functions inherited from mlir::detail::OpOrInterfaceRewritePatternBase< vector::ContractionOp >
LogicalResult	matchAndRewrite (Operation *op, PatternRewriter &rewriter) const final
	Wrapper around the RewritePattern method that passes the derived op type.
Public Member Functions inherited from mlir::RewritePattern
virtual	~RewritePattern ()=default
Public Member Functions inherited from mlir::Pattern
ArrayRef< OperationName >	getGeneratedOps () const
	Return a list of operations that may be generated when rewriting an operation instance with this pattern.
std::optional< OperationName >	getRootKind () const
	Return the root node that this pattern matches.
std::optional< TypeID >	getRootInterfaceID () const
	Return the interface ID used to match the root operation of this pattern.
std::optional< TypeID >	getRootTraitID () const
	Return the trait ID used to match the root operation of this pattern.
PatternBenefit	getBenefit () const
	Return the benefit (the inverse of "cost") of matching this pattern.
bool	hasBoundedRewriteRecursion () const
	Returns true if this pattern is known to result in recursive application, i.e.
MLIRContext *	getContext () const
	Return the MLIRContext used to create this pattern.
StringRef	getDebugName () const
	Return a readable name for this pattern.
void	setDebugName (StringRef name)
	Set the human readable debug name used for this pattern.
ArrayRef< StringRef >	getDebugLabels () const
	Return the set of debug labels attached to this pattern.
void	addDebugLabels (ArrayRef< StringRef > labels)
	Add the provided debug labels to this pattern.
void	addDebugLabels (StringRef label)

Static Public Member Functions
static LogicalResult	defaultFilter (vector::ContractionOp op)
Static Public Member Functions inherited from mlir::RewritePattern
template<typename T, typename... Args>
static std::unique_ptr< T >	create (Args &&...args)
	This method provides a convenient interface for creating and initializing derived rewrite patterns of the given type T.

Additional Inherited Members
Protected Member Functions inherited from mlir::RewritePattern
	Pattern (StringRef rootName, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
	Inherit the base constructors from Pattern.
	Pattern (MatchAnyOpTypeTag tag, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
	Inherit the base constructors from Pattern.
	Pattern (MatchInterfaceOpTypeTag tag, TypeID interfaceID, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
	Inherit the base constructors from Pattern.
	Pattern (MatchTraitOpTypeTag tag, TypeID traitID, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
	Inherit the base constructors from Pattern.
Protected Member Functions inherited from mlir::Pattern
	Pattern (StringRef rootName, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
	Construct a pattern with a certain benefit that matches the operation with the given root name.
	Pattern (MatchAnyOpTypeTag tag, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
	Construct a pattern that may match any operation type.
	Pattern (MatchInterfaceOpTypeTag tag, TypeID interfaceID, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
	Construct a pattern that may match any operation that implements the interface defined by the provided interfaceID.
	Pattern (MatchTraitOpTypeTag tag, TypeID traitID, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
	Construct a pattern that may match any operation that implements the trait defined by the provided traitID.
void	setHasBoundedRewriteRecursion (bool hasBoundedRecursionArg=true)
	Set the flag detailing if this pattern has bounded rewrite recursion or not.

Detailed Description

Lower vector.contract with all size one reduction dimensions to elementwise ops when possible.

Definition at line 740 of file LowerVectorContract.cpp.

Member Typedef Documentation

◆ FilterConstraintType

using ContractOpToElementwise::FilterConstraintType

Initial value:

std::function<LogicalResult(vector::ContractionOp op)>

Definition at line 743 of file LowerVectorContract.cpp.

Constructor & Destructor Documentation

◆ ContractOpToElementwise()

ContractOpToElementwise::ContractOpToElementwise	(	vector::VectorContractLowering	vectorContractLowering,
		MLIRContext *	context,
		PatternBenefit	benefit = 1,
		const FilterConstraintType &	constraint = defaultFilter )

inline

Definition at line 748 of file LowerVectorContract.cpp.

References defaultFilter().

Member Function Documentation

◆ defaultFilter()

LogicalResult ContractOpToElementwise::defaultFilter ( vector::ContractionOp op )

inlinestatic

Definition at line 745 of file LowerVectorContract.cpp.

References success().

Referenced by ContractOpToElementwise().

◆ matchAndRewriteMaskableOp()

FailureOr< Value > ContractOpToElementwise::matchAndRewriteMaskableOp	(	vector::ContractionOp	contractOp,
		MaskingOpInterface	maskOp,
		PatternRewriter &	rewriter ) const

inlineoverridevirtual

Implements mlir::vector::MaskableOpRewritePattern< vector::ContractionOp >.

Definition at line 756 of file LowerVectorContract.cpp.

References createContractArithOp(), getDimPosition(), mlir::AffineMap::getDimPosition(), mlir::AffineMap::getNumResults(), getReductionIndex(), and result.

The documentation for this struct was generated from the following file:

lib/Dialect/Vector/Transforms/LowerVectorContract.cpp

Public Types

Public Member Functions

Static Public Member Functions

Additional Inherited Members