MLIR  20.0.0git
Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
mlir::ConversionPattern Class Reference

Base class for the conversion patterns. More...

#include "mlir/Transforms/DialectConversion.h"

+ Inheritance diagram for mlir::ConversionPattern:

Public Member Functions

virtual void rewrite (Operation *op, ArrayRef< Value > operands, ConversionPatternRewriter &rewriter) const
 Hook for derived classes to implement rewriting. More...
 
virtual void rewrite (Operation *op, ArrayRef< ValueRange > operands, ConversionPatternRewriter &rewriter) const
 
virtual LogicalResult matchAndRewrite (Operation *op, ArrayRef< Value > operands, ConversionPatternRewriter &rewriter) const
 Hook for derived classes to implement combined matching and rewriting. More...
 
virtual LogicalResult matchAndRewrite (Operation *op, ArrayRef< ValueRange > operands, ConversionPatternRewriter &rewriter) const
 Hook for derived classes to implement combined matching and rewriting. More...
 
LogicalResult matchAndRewrite (Operation *op, PatternRewriter &rewriter) const final
 Attempt to match and rewrite the IR root at the specified operation. More...
 
const TypeConvertergetTypeConverter () const
 Return the type converter held by this pattern, or nullptr if the pattern does not require type conversion. More...
 
template<typename ConverterTy >
std::enable_if_t< std::is_base_of< TypeConverter, ConverterTy >::value, const ConverterTy * > getTypeConverter () const
 
- Public Member Functions inherited from mlir::RewritePattern
virtual ~RewritePattern ()=default
 
virtual void rewrite (Operation *op, PatternRewriter &rewriter) const
 Rewrite the IR rooted at the specified operation with the result of this pattern, generating any new operations with the specified builder. More...
 
virtual LogicalResult match (Operation *op) const
 Attempt to match against code rooted at the specified operation, which is the same operation code as getRootKind(). More...
 
- Public Member Functions inherited from mlir::Pattern
ArrayRef< OperationNamegetGeneratedOps () const
 Return a list of operations that may be generated when rewriting an operation instance with this pattern. More...
 
std::optional< OperationNamegetRootKind () const
 Return the root node that this pattern matches. More...
 
std::optional< TypeIDgetRootInterfaceID () const
 Return the interface ID used to match the root operation of this pattern. More...
 
std::optional< TypeIDgetRootTraitID () const
 Return the trait ID used to match the root operation of this pattern. More...
 
PatternBenefit getBenefit () const
 Return the benefit (the inverse of "cost") of matching this pattern. More...
 
bool hasBoundedRewriteRecursion () const
 Returns true if this pattern is known to result in recursive application, i.e. More...
 
MLIRContextgetContext () const
 Return the MLIRContext used to create this pattern. More...
 
StringRef getDebugName () const
 Return a readable name for this pattern. More...
 
void setDebugName (StringRef name)
 Set the human readable debug name used for this pattern. More...
 
ArrayRef< StringRef > getDebugLabels () const
 Return the set of debug labels attached to this pattern. More...
 
void addDebugLabels (ArrayRef< StringRef > labels)
 Add the provided debug labels to this pattern. More...
 
void addDebugLabels (StringRef label)
 

Protected Member Functions

template<typename... Args>
 ConversionPattern (const TypeConverter &typeConverter, Args &&...args)
 Construct a conversion pattern with the given converter, and forward the remaining arguments to RewritePattern. More...
 
SmallVector< ValuegetOneToOneAdaptorOperands (ArrayRef< ValueRange > operands) const
 Given an array of value ranges, which are the inputs to a 1:N adaptor, try to extract the single value of each range to construct a the inputs for a 1:1 adaptor. More...
 
- Protected Member Functions inherited from mlir::RewritePattern
 Pattern (StringRef rootName, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
 Inherit the base constructors from Pattern. More...
 
 Pattern (MatchAnyOpTypeTag tag, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
 Inherit the base constructors from Pattern. More...
 
 Pattern (MatchInterfaceOpTypeTag tag, TypeID interfaceID, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
 Inherit the base constructors from Pattern. More...
 
 Pattern (MatchTraitOpTypeTag tag, TypeID traitID, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
 Inherit the base constructors from Pattern. More...
 
- 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. More...
 
 Pattern (MatchAnyOpTypeTag tag, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={})
 Construct a pattern that may match any operation type. More...
 
 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. More...
 
 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. More...
 
void setHasBoundedRewriteRecursion (bool hasBoundedRecursionArg=true)
 Set the flag detailing if this pattern has bounded rewrite recursion or not. More...
 

Protected Attributes

const TypeConvertertypeConverter = nullptr
 An optional type converter for use by this pattern. More...
 

Additional Inherited Members

- 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. More...
 

Detailed Description

Base class for the conversion patterns.

This pattern class enables type conversions, and other uses specific to the conversion framework. As such, patterns of this type can only be used with the 'apply*' methods below.

Definition at line 530 of file DialectConversion.h.

Constructor & Destructor Documentation

◆ ConversionPattern()

template<typename... Args>
mlir::ConversionPattern::ConversionPattern ( const TypeConverter typeConverter,
Args &&...  args 
)
inlineprotected

Construct a conversion pattern with the given converter, and forward the remaining arguments to RewritePattern.

Definition at line 589 of file DialectConversion.h.

Member Function Documentation

◆ getOneToOneAdaptorOperands()

SmallVector< Value > ConversionPattern::getOneToOneAdaptorOperands ( ArrayRef< ValueRange operands) const
protected

Given an array of value ranges, which are the inputs to a 1:N adaptor, try to extract the single value of each range to construct a the inputs for a 1:1 adaptor.

This function produces a fatal error if at least one range has 0 or more than 1 value: "pattern 'name' does not support 1:N conversion"

Definition at line 1851 of file DialectConversion.cpp.

Referenced by matchAndRewrite(), mlir::OpInterfaceConversionPattern< SourceOp >::matchAndRewrite(), mlir::ConvertOpToLLVMPattern< SourceOp >::matchAndRewrite(), mlir::OpConversionPattern< SourceOp >::matchAndRewrite(), rewrite(), mlir::OpInterfaceConversionPattern< SourceOp >::rewrite(), mlir::ConvertOpToLLVMPattern< SourceOp >::rewrite(), and mlir::OpConversionPattern< SourceOp >::rewrite().

◆ getTypeConverter() [1/2]

const TypeConverter* mlir::ConversionPattern::getTypeConverter ( ) const
inline

Return the type converter held by this pattern, or nullptr if the pattern does not require type conversion.

Definition at line 573 of file DialectConversion.h.

References typeConverter.

Referenced by mlir::ConvertToLLVMPattern::getTypeConverter(), and mlir::spirv::ElementwiseOpPattern< Op, SPIRVOp >::matchAndRewrite().

◆ getTypeConverter() [2/2]

template<typename ConverterTy >
std::enable_if_t<std::is_base_of<TypeConverter, ConverterTy>::value, const ConverterTy *> mlir::ConversionPattern::getTypeConverter ( ) const
inline

Definition at line 578 of file DialectConversion.h.

References typeConverter.

◆ matchAndRewrite() [1/3]

virtual LogicalResult mlir::ConversionPattern::matchAndRewrite ( Operation op,
ArrayRef< Value operands,
ConversionPatternRewriter rewriter 
) const
inlinevirtual

◆ matchAndRewrite() [2/3]

virtual LogicalResult mlir::ConversionPattern::matchAndRewrite ( Operation op,
ArrayRef< ValueRange operands,
ConversionPatternRewriter rewriter 
) const
inlinevirtual

◆ matchAndRewrite() [3/3]

LogicalResult ConversionPattern::matchAndRewrite ( Operation op,
PatternRewriter rewriter 
) const
finalvirtual

◆ rewrite() [1/2]

virtual void mlir::ConversionPattern::rewrite ( Operation op,
ArrayRef< Value operands,
ConversionPatternRewriter rewriter 
) const
inlinevirtual

◆ rewrite() [2/2]

virtual void mlir::ConversionPattern::rewrite ( Operation op,
ArrayRef< ValueRange operands,
ConversionPatternRewriter rewriter 
) const
inlinevirtual

Member Data Documentation

◆ typeConverter

const TypeConverter* mlir::ConversionPattern::typeConverter = nullptr
protected

The documentation for this class was generated from the following files: