13 #ifndef MLIR_TRANSFORMS_DIALECTCONVERSION_H_
14 #define MLIR_TRANSFORMS_DIALECTCONVERSION_H_
16 #include "mlir/Config/mlir-config.h"
18 #include "llvm/ADT/MapVector.h"
19 #include "llvm/ADT/StringMap.h"
20 #include <type_traits>
27 struct ConversionConfig;
28 class ConversionPatternRewriter;
31 struct OperationConverter;
47 : conversions(other.conversions),
48 sourceMaterializations(other.sourceMaterializations),
49 targetMaterializations(other.targetMaterializations),
50 typeAttributeConversions(other.typeAttributeConversions) {}
52 conversions = other.conversions;
53 sourceMaterializations = other.sourceMaterializations;
54 targetMaterializations = other.targetMaterializations;
55 typeAttributeConversions = other.typeAttributeConversions;
64 : remappedInputs(numOrigInputs) {}
81 return remappedInputs[input];
103 void remapInput(
unsigned origInputNo,
unsigned newInputNo,
104 unsigned newInputCount = 1);
133 llvm::PointerIntPair<Attribute, 2>
impl;
136 static constexpr
unsigned naTag = 0;
137 static constexpr
unsigned resultTag = 1;
138 static constexpr
unsigned abortTag = 2;
159 template <
typename FnT,
typename T =
typename llvm::function_traits<
160 std::decay_t<FnT>>::template arg_t<0>>
162 registerConversion(wrapCallback<T>(std::forward<FnT>(callback)));
185 template <
typename FnT,
typename T =
typename llvm::function_traits<
186 std::decay_t<FnT>>::template arg_t<1>>
188 sourceMaterializations.emplace_back(
189 wrapSourceMaterialization<T>(std::forward<FnT>(callback)));
209 template <
typename FnT,
typename T =
typename llvm::function_traits<
210 std::decay_t<FnT>>::template arg_t<1>>
212 targetMaterializations.emplace_back(
213 wrapTargetMaterialization<T>(std::forward<FnT>(callback)));
236 typename llvm::function_traits<std::decay_t<FnT>>::template arg_t<0>,
238 typename llvm::function_traits<std::decay_t<FnT>>::template arg_t<1>>
240 registerTypeAttributeConversion(
241 wrapTypeAttributeConversion<T, A>(std::forward<FnT>(callback)));
257 template <
typename TargetType>
259 return dyn_cast_or_null<TargetType>(
convertType(t));
273 template <
typename RangeT>
274 std::enable_if_t<!std::is_convertible<RangeT, Type>::value &&
275 !std::is_convertible<RangeT, Operation *>::value,
278 return llvm::all_of(range, [
this](
Type type) {
return isLegal(type); });
294 SignatureConversion &result)
const;
296 SignatureConversion &result,
297 unsigned origInputOffset = 0)
const;
312 Type originalType = {})
const;
315 TypeRange resultType,
317 Type originalType = {})
const;
324 Attribute attr)
const;
330 using ConversionCallbackFn = std::function<std::optional<LogicalResult>(
331 Type, SmallVectorImpl<Type> &)>;
336 using SourceMaterializationCallbackFn =
337 std::function<
Value(OpBuilder &, Type, ValueRange, Location)>;
342 using TargetMaterializationCallbackFn = std::function<SmallVector<Value>(
343 OpBuilder &, TypeRange, ValueRange, Location, Type)>;
346 using TypeAttributeConversionCallbackFn =
347 std::function<AttributeConversionResult(Type, Attribute)>;
352 template <
typename T,
typename FnT>
353 std::enable_if_t<std::is_invocable_v<FnT, T>, ConversionCallbackFn>
354 wrapCallback(FnT &&callback)
const {
355 return wrapCallback<T>([callback = std::forward<FnT>(callback)](
356 T type, SmallVectorImpl<Type> &results) {
357 if (std::optional<Type> resultOpt = callback(type)) {
358 bool wasSuccess =
static_cast<bool>(*resultOpt);
360 results.push_back(*resultOpt);
361 return std::optional<LogicalResult>(success(wasSuccess));
363 return std::optional<LogicalResult>();
368 template <
typename T,
typename FnT>
369 std::enable_if_t<std::is_invocable_v<FnT, T, SmallVectorImpl<Type> &>,
370 ConversionCallbackFn>
371 wrapCallback(FnT &&callback)
const {
372 return [callback = std::forward<FnT>(callback)](
374 SmallVectorImpl<Type> &results) -> std::optional<LogicalResult> {
375 T derivedType = dyn_cast<T>(type);
378 return callback(derivedType, results);
383 void registerConversion(ConversionCallbackFn callback) {
384 conversions.emplace_back(std::move(callback));
385 cachedDirectConversions.clear();
386 cachedMultiConversions.clear();
392 template <
typename T,
typename FnT>
393 SourceMaterializationCallbackFn
394 wrapSourceMaterialization(FnT &&callback)
const {
395 return [callback = std::forward<FnT>(callback)](
396 OpBuilder &builder,
Type resultType, ValueRange inputs,
397 Location loc) -> Value {
398 if (T derivedType = dyn_cast<T>(resultType))
399 return callback(builder, derivedType, inputs, loc);
412 template <
typename T,
typename FnT>
414 std::is_invocable_v<FnT, OpBuilder &, T, ValueRange, Location, Type>,
415 TargetMaterializationCallbackFn>
416 wrapTargetMaterialization(FnT &&callback)
const {
417 return [callback = std::forward<FnT>(callback)](
418 OpBuilder &builder, TypeRange resultTypes, ValueRange inputs,
419 Location loc,
Type originalType) -> SmallVector<Value> {
420 SmallVector<Value> result;
421 if constexpr (std::is_same<T, TypeRange>::value) {
424 result = callback(builder, resultTypes, inputs, loc, originalType);
425 }
else if constexpr (std::is_assignable<Type, T>::value) {
428 if (resultTypes.size() == 1) {
431 if (T derivedType = dyn_cast<T>(resultTypes.front())) {
436 callback(builder, derivedType, inputs, loc, originalType);
438 result.push_back(val);
442 static_assert(
sizeof(T) == 0,
"T must be a Type or a TypeRange");
450 template <
typename T,
typename FnT>
452 std::is_invocable_v<FnT, OpBuilder &, T, ValueRange, Location>,
453 TargetMaterializationCallbackFn>
454 wrapTargetMaterialization(FnT &&callback)
const {
455 return wrapTargetMaterialization<T>(
456 [callback = std::forward<FnT>(callback)](
457 OpBuilder &builder, T resultTypes, ValueRange inputs, Location loc,
459 return callback(builder, resultTypes, inputs, loc);
467 template <
typename T,
typename A,
typename FnT>
468 TypeAttributeConversionCallbackFn
469 wrapTypeAttributeConversion(FnT &&callback)
const {
470 return [callback = std::forward<FnT>(callback)](
471 Type type, Attribute attr) -> AttributeConversionResult {
472 if (T derivedType = dyn_cast<T>(type)) {
473 if (A derivedAttr = dyn_cast_or_null<A>(attr))
474 return callback(derivedType, derivedAttr);
482 registerTypeAttributeConversion(TypeAttributeConversionCallbackFn callback) {
483 typeAttributeConversions.emplace_back(std::move(callback));
485 cachedDirectConversions.clear();
486 cachedMultiConversions.clear();
490 SmallVector<ConversionCallbackFn, 4> conversions;
493 SmallVector<SourceMaterializationCallbackFn, 2> sourceMaterializations;
494 SmallVector<TargetMaterializationCallbackFn, 2> targetMaterializations;
497 SmallVector<TypeAttributeConversionCallbackFn, 2> typeAttributeConversions;
502 mutable DenseMap<Type, Type> cachedDirectConversions;
504 mutable DenseMap<Type, SmallVector<Type, 2>> cachedMultiConversions;
506 mutable llvm::sys::SmartRWMutex<true> cacheMutex;
525 virtual LogicalResult
528 llvm_unreachable(
"matchAndRewrite is not implemented");
533 virtual LogicalResult
547 template <
typename ConverterTy>
548 std::enable_if_t<std::is_base_of<TypeConverter, ConverterTy>::value,
557 using RewritePattern::RewritePattern;
560 template <
typename... Args>
582 template <
typename SourceOp>
587 typename SourceOp::template GenericAdaptor<ArrayRef<ValueRange>>;
601 auto sourceOp = cast<SourceOp>(op);
607 auto sourceOp = cast<SourceOp>(op);
614 virtual LogicalResult
617 llvm_unreachable(
"matchAndRewrite is not implemented");
619 virtual LogicalResult
634 template <
typename SourceOp>
639 SourceOp::getInterfaceID(), benefit, context) {}
643 SourceOp::getInterfaceID(), benefit, context) {}
660 virtual LogicalResult
663 llvm_unreachable(
"matchAndRewrite is not implemented");
665 virtual LogicalResult
678 template <
template <
typename>
class TraitType>
683 TypeID::
get<TraitType>(), benefit, context) {}
687 TypeID::
get<TraitType>(), benefit, context) {}
693 FailureOr<Operation *>
695 const TypeConverter &converter,
696 ConversionPatternRewriter &rewriter);
702 StringRef functionLikeOpName, RewritePatternSet &
patterns,
703 const TypeConverter &converter);
705 template <
typename FuncOpT>
713 RewritePatternSet &
patterns,
const TypeConverter &converter);
720 struct ConversionPatternRewriterImpl;
807 template <
typename RangeT = ValueRange>
812 template <
typename RangeT>
815 ArrayRef(llvm::to_vector_of<ValueRange>(newValues)));
829 ValueRange argValues = std::nullopt)
override;
860 std::unique_ptr<detail::ConversionPatternRewriterImpl>
impl;
896 std::function<std::optional<bool>(
Operation *)>;
907 template <
typename OpT>
916 template <
typename OpT>
920 template <
typename OpT,
typename OpT2,
typename... OpTs>
931 setLegalityCallback(op, callback);
933 template <
typename OpT>
938 template <
typename OpT,
typename OpT2,
typename... OpTs>
940 addDynamicallyLegalOp<OpT>(callback);
943 template <
typename OpT,
class Callable>
944 std::enable_if_t<!std::is_invocable_v<Callable, Operation *>>
946 addDynamicallyLegalOp<OpT>(
947 [=](
Operation *op) {
return callback(cast<OpT>(op)); });
955 template <
typename OpT>
959 template <
typename OpT,
typename OpT2,
typename... OpTs>
972 template <
typename OpT>
977 template <
typename OpT,
typename OpT2,
typename... OpTs>
979 markOpRecursivelyLegal<OpT>(callback);
982 template <
typename OpT,
class Callable>
983 std::enable_if_t<!std::is_invocable_v<Callable, Operation *>>
985 markOpRecursivelyLegal<OpT>(
986 [=](
Operation *op) {
return callback(cast<OpT>(op)); });
994 template <
typename... Names>
999 template <
typename... Args>
1007 template <
typename... Names>
1009 StringRef name, Names... names) {
1012 setLegalityCallback(dialectNames, callback);
1014 template <
typename... Args>
1017 Args::getDialectNamespace()...);
1024 setLegalityCallback(fn);
1029 template <
typename... Names>
1034 template <
typename... Args>
1076 struct LegalizationInfo {
1081 bool isRecursivelyLegal =
false;
1088 std::optional<LegalizationInfo> getOpInfo(OperationName op)
const;
1092 llvm::MapVector<OperationName, LegalizationInfo> legalOperations;
1096 DenseMap<OperationName, DynamicLegalityCallbackFn> opRecursiveLegalityFns;
1100 llvm::StringMap<LegalizationAction> legalDialects;
1103 llvm::StringMap<DynamicLegalityCallbackFn> dialectLegalityFns;
1112 #if MLIR_ENABLE_PDL_IN_PATTERNMATCH
1148 class PDLConversionConfig final {
Attributes are known-constant values of operations.
This class represents an argument of a Block.
Block represents an ordered list of Operations.
OpListType::iterator iterator
This class implements a pattern rewriter for use with ConversionPatterns.
void replaceOp(Operation *op, ValueRange newValues) override
Replace the given operation with the new values.
LogicalResult getRemappedValues(ValueRange keys, SmallVectorImpl< Value > &results)
Return the converted values that replace 'keys' with types defined by the type converter of the curre...
FailureOr< Block * > convertRegionTypes(Region *region, const TypeConverter &converter, TypeConverter::SignatureConversion *entryConversion=nullptr)
Apply a signature conversion to each block in the given region.
void inlineBlockBefore(Block *source, Block *dest, Block::iterator before, ValueRange argValues=std::nullopt) override
PatternRewriter hook for inlining the ops of a block into another block.
void replaceOpWithMultiple(Operation *op, SmallVector< SmallVector< Value >> &&newValues)
Replace the given operation with the new value ranges.
Block * applySignatureConversion(Block *block, TypeConverter::SignatureConversion &conversion, const TypeConverter *converter=nullptr)
Apply a signature conversion to given block.
void replaceOpWithMultiple(Operation *op, ArrayRef< RangeT > newValues)
void startOpModification(Operation *op) override
PatternRewriter hook for updating the given operation in-place.
void eraseOp(Operation *op) override
PatternRewriter hook for erasing a dead operation.
void replaceOpWithMultiple(Operation *op, RangeT &&newValues)
detail::ConversionPatternRewriterImpl & getImpl()
Return a reference to the internal implementation.
void eraseBlock(Block *block) override
PatternRewriter hook for erase all operations in a block.
void cancelOpModification(Operation *op) override
PatternRewriter hook for updating the given operation in-place.
bool canRecoverFromRewriteFailure() const override
Indicate that the conversion rewriter can recover from rewrite failure.
Value getRemappedValue(Value key)
Return the converted value of 'key' with a type defined by the type converter of the currently execut...
void finalizeOpModification(Operation *op) override
PatternRewriter hook for updating the given operation in-place.
void replaceUsesOfBlockArgument(BlockArgument from, Value to)
Replace all the uses of the block argument from with value to.
~ConversionPatternRewriter() override
Base class for the conversion patterns.
SmallVector< Value > getOneToOneAdaptorOperands(ArrayRef< ValueRange > operands) const
Given an array of value ranges, which are the inputs to a 1:N adaptor, try to extract the single valu...
const TypeConverter * typeConverter
An optional type converter for use by this pattern.
ConversionPattern(const TypeConverter &typeConverter, Args &&...args)
Construct a conversion pattern with the given converter, and forward the remaining arguments to Rewri...
const TypeConverter * getTypeConverter() const
Return the type converter held by this pattern, or nullptr if the pattern does not require type conve...
std::enable_if_t< std::is_base_of< TypeConverter, ConverterTy >::value, const ConverterTy * > getTypeConverter() const
virtual LogicalResult matchAndRewrite(Operation *op, ArrayRef< Value > operands, ConversionPatternRewriter &rewriter) const
Hook for derived classes to implement combined matching and rewriting.
virtual LogicalResult matchAndRewrite(Operation *op, ArrayRef< ValueRange > operands, ConversionPatternRewriter &rewriter) const
Hook for derived classes to implement combined matching and rewriting.
This class describes a specific conversion target.
void setDialectAction(ArrayRef< StringRef > dialectNames, LegalizationAction action)
Register a legality action for the given dialects.
void setOpAction(LegalizationAction action)
void addLegalOp(OperationName op)
Register the given operations as legal.
void addDynamicallyLegalDialect(DynamicLegalityCallbackFn callback)
void setOpAction(OperationName op, LegalizationAction action)
Register a legality action for the given operation.
void addDynamicallyLegalDialect(const DynamicLegalityCallbackFn &callback, StringRef name, Names... names)
Register the operations of the given dialects as dynamically legal, i.e.
void addDynamicallyLegalOp(const DynamicLegalityCallbackFn &callback)
void addLegalDialect(StringRef name, Names... names)
Register the operations of the given dialects as legal.
std::optional< LegalOpDetails > isLegal(Operation *op) const
If the given operation instance is legal on this target, a structure containing legality information ...
std::enable_if_t<!std::is_invocable_v< Callable, Operation * > > markOpRecursivelyLegal(Callable &&callback)
void markUnknownOpDynamicallyLegal(const DynamicLegalityCallbackFn &fn)
Register unknown operations as dynamically legal.
std::optional< LegalizationAction > getOpAction(OperationName op) const
Get the legality action for the given operation.
void addDynamicallyLegalOp(OperationName op, const DynamicLegalityCallbackFn &callback)
Register the given operation as dynamically legal and set the dynamic legalization callback to the on...
void addIllegalDialect(StringRef name, Names... names)
Register the operations of the given dialects as illegal, i.e.
std::enable_if_t<!std::is_invocable_v< Callable, Operation * > > addDynamicallyLegalOp(Callable &&callback)
void addDynamicallyLegalOp(const DynamicLegalityCallbackFn &callback)
void markOpRecursivelyLegal(const DynamicLegalityCallbackFn &callback={})
void addIllegalOp(OperationName op)
Register the given operation as illegal, i.e.
LegalizationAction
This enumeration corresponds to the specific action to take when considering an operation legal for t...
@ Illegal
The target explicitly does not support this operation.
@ Dynamic
This operation has dynamic legalization constraints that must be checked by the target.
@ Legal
The target supports this operation.
ConversionTarget(MLIRContext &ctx)
void markOpRecursivelyLegal(OperationName name, const DynamicLegalityCallbackFn &callback)
Mark an operation, that must have either been set as Legal or DynamicallyLegal, as being recursively ...
void markOpRecursivelyLegal(const DynamicLegalityCallbackFn &callback={})
std::function< std::optional< bool >(Operation *)> DynamicLegalityCallbackFn
The signature of the callback used to determine if an operation is dynamically legal on the target.
bool isIllegal(Operation *op) const
Returns true is operation instance is illegal on this target.
virtual ~ConversionTarget()=default
This class contains all of the information necessary to report a diagnostic to the DiagnosticEngine.
This class represents a frozen set of patterns that can be processed by a pattern applicator.
This class defines the main interface for locations in MLIR and acts as a non-nullable wrapper around...
MLIRContext is the top-level object for a collection of MLIR operations.
This class helps build Operations.
void setListener(Listener *newListener)
Sets the listener of this builder to the one provided.
OpConversionPattern is a wrapper around ConversionPattern that allows for matching and rewriting agai...
LogicalResult matchAndRewrite(Operation *op, ArrayRef< ValueRange > operands, ConversionPatternRewriter &rewriter) const final
Hook for derived classes to implement combined matching and rewriting.
typename SourceOp::Adaptor OpAdaptor
virtual LogicalResult matchAndRewrite(SourceOp op, OneToNOpAdaptor adaptor, ConversionPatternRewriter &rewriter) const
OpConversionPattern(MLIRContext *context, PatternBenefit benefit=1)
typename SourceOp::template GenericAdaptor< ArrayRef< ValueRange > > OneToNOpAdaptor
OpConversionPattern(const TypeConverter &typeConverter, MLIRContext *context, PatternBenefit benefit=1)
virtual LogicalResult matchAndRewrite(SourceOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const
Methods that operate on the SourceOp type.
LogicalResult matchAndRewrite(Operation *op, ArrayRef< Value > operands, ConversionPatternRewriter &rewriter) const final
Wrappers around the ConversionPattern methods that pass the derived op type.
OpInterfaceConversionPattern is a wrapper around ConversionPattern that allows for matching and rewri...
LogicalResult matchAndRewrite(Operation *op, ArrayRef< ValueRange > operands, ConversionPatternRewriter &rewriter) const final
Hook for derived classes to implement combined matching and rewriting.
virtual LogicalResult matchAndRewrite(SourceOp op, ArrayRef< ValueRange > operands, ConversionPatternRewriter &rewriter) const
LogicalResult matchAndRewrite(Operation *op, ArrayRef< Value > operands, ConversionPatternRewriter &rewriter) const final
Wrappers around the ConversionPattern methods that pass the derived op type.
OpInterfaceConversionPattern(MLIRContext *context, PatternBenefit benefit=1)
OpInterfaceConversionPattern(const TypeConverter &typeConverter, MLIRContext *context, PatternBenefit benefit=1)
virtual LogicalResult matchAndRewrite(SourceOp op, ArrayRef< Value > operands, ConversionPatternRewriter &rewriter) const
Methods that operate on the SourceOp type.
OpTraitConversionPattern is a wrapper around ConversionPattern that allows for matching and rewriting...
OpTraitConversionPattern(const TypeConverter &typeConverter, MLIRContext *context, PatternBenefit benefit=1)
OpTraitConversionPattern(MLIRContext *context, PatternBenefit benefit=1)
Operation is the basic unit of execution within MLIR.
A PDL configuration that is used to supported dialect conversion functionality.
void notifyRewriteEnd(PatternRewriter &rewriter) final
void notifyRewriteBegin(PatternRewriter &rewriter) final
Hooks that are invoked at the beginning and end of a rewrite of a matched pattern.
const TypeConverter * getTypeConverter() const
Return the type converter used by this configuration, which may be nullptr if no type conversions are...
PDLConversionConfig(const TypeConverter *converter)
~PDLConversionConfig() final=default
This class represents the benefit of a pattern match in a unitless scheme that ranges from 0 (very li...
A special type of RewriterBase that coordinates the application of a rewrite pattern on the current I...
This class contains all of the data related to a pattern, but does not contain any methods or logic f...
This class contains a list of basic blocks and a link to the parent operation it is attached to.
RewritePattern is the common base class for all DAG to DAG replacements.
virtual void inlineBlockBefore(Block *source, Block *dest, Block::iterator before, ValueRange argValues=std::nullopt)
Inline the operations of block 'source' into block 'dest' before the given position.
The general result of a type attribute conversion callback, allowing for early termination.
Attribute getResult() const
constexpr AttributeConversionResult()
static AttributeConversionResult abort()
static AttributeConversionResult na()
AttributeConversionResult(Attribute attr)
static AttributeConversionResult result(Attribute attr)
This class provides all of the information necessary to convert a type signature.
void addInputs(unsigned origInputNo, ArrayRef< Type > types)
Remap an input of the original signature with a new set of types.
std::optional< InputMapping > getInputMapping(unsigned input) const
Get the input mapping for the given argument.
ArrayRef< Type > getConvertedTypes() const
Return the argument types for the new signature.
void remapInput(unsigned origInputNo, ArrayRef< Value > replacements)
Remap an input of the original signature to replacements values.
SignatureConversion(unsigned numOrigInputs)
std::optional< Attribute > convertTypeAttribute(Type type, Attribute attr) const
Convert an attribute present attr from within the type type using the registered conversion functions...
Value materializeSourceConversion(OpBuilder &builder, Location loc, Type resultType, ValueRange inputs) const
Materialize a conversion from a set of types into one result type by generating a cast sequence of so...
void addConversion(FnT &&callback)
Register a conversion function.
bool isLegal(Type type) const
Return true if the given type is legal for this type converter, i.e.
LogicalResult convertSignatureArgs(TypeRange types, SignatureConversion &result, unsigned origInputOffset=0) const
std::enable_if_t<!std::is_convertible< RangeT, Type >::value &&!std::is_convertible< RangeT, Operation * >::value, bool > isLegal(RangeT &&range) const
Return true if all of the given types are legal for this type converter.
TargetType convertType(Type t) const
Attempts a 1-1 type conversion, expecting the result type to be TargetType.
LogicalResult convertSignatureArg(unsigned inputNo, Type type, SignatureConversion &result) const
This method allows for converting a specific argument of a signature.
TypeConverter(const TypeConverter &other)
TypeConverter & operator=(const TypeConverter &other)
LogicalResult convertType(Type t, SmallVectorImpl< Type > &results) const
Convert the given type.
std::optional< SignatureConversion > convertBlockSignature(Block *block) const
This function converts the type signature of the given block, by invoking 'convertSignatureArg' for e...
void addSourceMaterialization(FnT &&callback)
All of the following materializations require function objects that are convertible to the following ...
virtual ~TypeConverter()=default
void addTypeAttributeConversion(FnT &&callback)
Register a conversion function for attributes within types.
void addTargetMaterialization(FnT &&callback)
This method registers a materialization that will be called when converting a value to a target type ...
LogicalResult convertTypes(TypeRange types, SmallVectorImpl< Type > &results) const
Convert the given set of types, filling 'results' as necessary.
bool isSignatureLegal(FunctionType ty) const
Return true if the inputs and outputs of the given function type are legal.
Value materializeTargetConversion(OpBuilder &builder, Location loc, Type resultType, ValueRange inputs, Type originalType={}) const
This class provides an efficient unique identifier for a specific C++ type.
This class provides an abstraction over the various different ranges of value types.
Instances of the Type class are uniqued, have an immutable identifier and an optional mutable compone...
This class provides an abstraction over the different types of ranges over Values.
This class represents an instance of an SSA value in the MLIR system, representing a computable value...
@ Type
An inlay hint that for a type annotation.
Include the generated interface declarations.
void populateFunctionOpInterfaceTypeConversionPattern(StringRef functionLikeOpName, RewritePatternSet &patterns, const TypeConverter &converter)
Add a pattern to the given pattern list to convert the signature of a FunctionOpInterface op with the...
void populateAnyFunctionOpInterfaceTypeConversionPattern(RewritePatternSet &patterns, const TypeConverter &converter)
const FrozenRewritePatternSet GreedyRewriteConfig config
LogicalResult applyFullConversion(ArrayRef< Operation * > ops, const ConversionTarget &target, const FrozenRewritePatternSet &patterns, ConversionConfig config=ConversionConfig())
Apply a complete conversion on the given operations, and all nested operations.
FailureOr< Operation * > convertOpResultTypes(Operation *op, ValueRange operands, const TypeConverter &converter, ConversionPatternRewriter &rewriter)
Generic utility to convert op result types according to type converter without knowing exact op type.
const FrozenRewritePatternSet & patterns
LogicalResult applyAnalysisConversion(ArrayRef< Operation * > ops, ConversionTarget &target, const FrozenRewritePatternSet &patterns, ConversionConfig config=ConversionConfig())
Apply an analysis conversion on the given operations, and all nested operations.
void reconcileUnrealizedCasts(ArrayRef< UnrealizedConversionCastOp > castOps, SmallVectorImpl< UnrealizedConversionCastOp > *remainingCastOps=nullptr)
Try to reconcile all given UnrealizedConversionCastOps and store the left-over ops in remainingCastOp...
auto get(MLIRContext *context, Ts &&...params)
Helper method that injects context only if needed, this helps unify some of the attribute constructio...
void registerConversionPDLFunctions(RewritePatternSet &patterns)
Register the dialect conversion PDL functions with the given pattern set.
LogicalResult applyPartialConversion(ArrayRef< Operation * > ops, const ConversionTarget &target, const FrozenRewritePatternSet &patterns, ConversionConfig config=ConversionConfig())
Below we define several entry points for operation conversion.
Dialect conversion configuration.
bool allowPatternRollback
If set to "true", pattern rollback is allowed.
RewriterBase::Listener * listener
An optional listener that is notified about all IR modifications in case dialect conversion succeeds.
function_ref< void(Diagnostic &)> notifyCallback
An optional callback used to notify about match failure diagnostics during the conversion.
DenseSet< Operation * > * legalizableOps
Analysis conversion only.
DenseSet< Operation * > * unlegalizedOps
Partial conversion only.
bool buildMaterializations
If set to "true", the dialect conversion attempts to build source/target materializations through the...
A structure containing additional information describing a specific legal operation instance.
bool isRecursivelyLegal
A flag that indicates if this operation is 'recursively' legal.
This class acts as a special tag that makes the desire to match any operation that implements a given...
This class acts as a special tag that makes the desire to match any operation that implements a given...