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)));
237 typename llvm::function_traits<std::decay_t<FnT>>::template arg_t<0>,
239 typename llvm::function_traits<std::decay_t<FnT>>::template arg_t<1>>
241 registerTypeAttributeConversion(
242 wrapTypeAttributeConversion<T, A>(std::forward<FnT>(callback)));
258 template <
typename TargetType>
260 return dyn_cast_or_null<TargetType>(
convertType(t));
274 template <
typename RangeT>
275 std::enable_if_t<!std::is_convertible<RangeT, Type>::value &&
276 !std::is_convertible<RangeT, Operation *>::value,
279 return llvm::all_of(range, [
this](
Type type) {
return isLegal(type); });
295 SignatureConversion &result)
const;
297 SignatureConversion &result,
298 unsigned origInputOffset = 0)
const;
313 Type originalType = {})
const;
316 TypeRange resultType,
318 Type originalType = {})
const;
325 Attribute attr)
const;
331 using ConversionCallbackFn = std::function<std::optional<LogicalResult>(
332 Type, SmallVectorImpl<Type> &)>;
337 using SourceMaterializationCallbackFn =
338 std::function<
Value(OpBuilder &, Type, ValueRange, Location)>;
343 using TargetMaterializationCallbackFn = std::function<SmallVector<Value>(
344 OpBuilder &, TypeRange, ValueRange, Location, Type)>;
347 using TypeAttributeConversionCallbackFn =
348 std::function<AttributeConversionResult(Type, Attribute)>;
353 template <
typename T,
typename FnT>
354 std::enable_if_t<std::is_invocable_v<FnT, T>, ConversionCallbackFn>
355 wrapCallback(FnT &&callback)
const {
356 return wrapCallback<T>([callback = std::forward<FnT>(callback)](
357 T type, SmallVectorImpl<Type> &results) {
358 if (std::optional<Type> resultOpt = callback(type)) {
359 bool wasSuccess =
static_cast<bool>(*resultOpt);
361 results.push_back(*resultOpt);
362 return std::optional<LogicalResult>(success(wasSuccess));
364 return std::optional<LogicalResult>();
369 template <
typename T,
typename FnT>
370 std::enable_if_t<std::is_invocable_v<FnT, T, SmallVectorImpl<Type> &>,
371 ConversionCallbackFn>
372 wrapCallback(FnT &&callback)
const {
373 return [callback = std::forward<FnT>(callback)](
375 SmallVectorImpl<Type> &results) -> std::optional<LogicalResult> {
376 T derivedType = dyn_cast<T>(type);
379 return callback(derivedType, results);
384 void registerConversion(ConversionCallbackFn callback) {
385 conversions.emplace_back(std::move(callback));
386 cachedDirectConversions.clear();
387 cachedMultiConversions.clear();
393 template <
typename T,
typename FnT>
394 SourceMaterializationCallbackFn
395 wrapSourceMaterialization(FnT &&callback)
const {
396 return [callback = std::forward<FnT>(callback)](
397 OpBuilder &builder,
Type resultType, ValueRange inputs,
398 Location loc) -> Value {
399 if (T derivedType = dyn_cast<T>(resultType))
400 return callback(builder, derivedType, inputs, loc);
413 template <
typename T,
typename FnT>
415 std::is_invocable_v<FnT, OpBuilder &, T, ValueRange, Location, Type>,
416 TargetMaterializationCallbackFn>
417 wrapTargetMaterialization(FnT &&callback)
const {
418 return [callback = std::forward<FnT>(callback)](
419 OpBuilder &builder, TypeRange resultTypes, ValueRange inputs,
420 Location loc,
Type originalType) -> SmallVector<Value> {
421 SmallVector<Value> result;
422 if constexpr (std::is_same<T, TypeRange>::value) {
425 result = callback(builder, resultTypes, inputs, loc, originalType);
426 }
else if constexpr (std::is_assignable<Type, T>::value) {
429 if (resultTypes.size() == 1) {
432 if (T derivedType = dyn_cast<T>(resultTypes.front())) {
437 callback(builder, derivedType, inputs, loc, originalType);
439 result.push_back(val);
443 static_assert(
sizeof(T) == 0,
"T must be a Type or a TypeRange");
451 template <
typename T,
typename FnT>
453 std::is_invocable_v<FnT, OpBuilder &, T, ValueRange, Location>,
454 TargetMaterializationCallbackFn>
455 wrapTargetMaterialization(FnT &&callback)
const {
456 return wrapTargetMaterialization<T>(
457 [callback = std::forward<FnT>(callback)](
458 OpBuilder &builder, T resultTypes, ValueRange inputs, Location loc,
460 return callback(builder, resultTypes, inputs, loc);
468 template <
typename T,
typename A,
typename FnT>
469 TypeAttributeConversionCallbackFn
470 wrapTypeAttributeConversion(FnT &&callback)
const {
471 return [callback = std::forward<FnT>(callback)](
472 Type type, Attribute attr) -> AttributeConversionResult {
473 if (T derivedType = dyn_cast<T>(type)) {
474 if (A derivedAttr = dyn_cast_or_null<A>(attr))
475 return callback(derivedType, derivedAttr);
483 registerTypeAttributeConversion(TypeAttributeConversionCallbackFn callback) {
484 typeAttributeConversions.emplace_back(std::move(callback));
486 cachedDirectConversions.clear();
487 cachedMultiConversions.clear();
491 SmallVector<ConversionCallbackFn, 4> conversions;
494 SmallVector<SourceMaterializationCallbackFn, 2> sourceMaterializations;
495 SmallVector<TargetMaterializationCallbackFn, 2> targetMaterializations;
498 SmallVector<TypeAttributeConversionCallbackFn, 2> typeAttributeConversions;
503 mutable DenseMap<Type, Type> cachedDirectConversions;
505 mutable DenseMap<Type, SmallVector<Type, 2>> cachedMultiConversions;
507 mutable llvm::sys::SmartRWMutex<true> cacheMutex;
526 virtual LogicalResult
529 llvm_unreachable(
"matchAndRewrite is not implemented");
534 virtual LogicalResult
548 template <
typename ConverterTy>
549 std::enable_if_t<std::is_base_of<TypeConverter, ConverterTy>::value,
558 using RewritePattern::RewritePattern;
561 template <
typename... Args>
583 template <
typename SourceOp>
588 typename SourceOp::template GenericAdaptor<ArrayRef<ValueRange>>;
602 auto sourceOp = cast<SourceOp>(op);
608 auto sourceOp = cast<SourceOp>(op);
615 virtual LogicalResult
618 llvm_unreachable(
"matchAndRewrite is not implemented");
620 virtual LogicalResult
635 template <
typename SourceOp>
640 SourceOp::getInterfaceID(), benefit, context) {}
644 SourceOp::getInterfaceID(), benefit, context) {}
661 virtual LogicalResult
664 llvm_unreachable(
"matchAndRewrite is not implemented");
666 virtual LogicalResult
679 template <
template <
typename>
class TraitType>
684 TypeID::
get<TraitType>(), benefit, context) {}
688 TypeID::
get<TraitType>(), benefit, context) {}
694 FailureOr<Operation *>
696 const TypeConverter &converter,
697 ConversionPatternRewriter &rewriter);
703 StringRef functionLikeOpName, RewritePatternSet &
patterns,
704 const TypeConverter &converter);
706 template <
typename FuncOpT>
714 RewritePatternSet &
patterns,
const TypeConverter &converter);
721 struct ConversionPatternRewriterImpl;
809 template <
typename RangeT = ValueRange>
814 template <
typename RangeT>
817 ArrayRef(llvm::to_vector_of<ValueRange>(newValues)));
847 detail::ConversionPatternRewriterImpl &
getImpl();
862 std::unique_ptr<detail::ConversionPatternRewriterImpl>
impl;
898 std::function<std::optional<bool>(
Operation *)>;
909 template <
typename OpT>
918 template <
typename OpT>
922 template <
typename OpT,
typename OpT2,
typename... OpTs>
933 setLegalityCallback(op, callback);
935 template <
typename OpT>
940 template <
typename OpT,
typename OpT2,
typename... OpTs>
942 addDynamicallyLegalOp<OpT>(callback);
945 template <
typename OpT,
class Callable>
946 std::enable_if_t<!std::is_invocable_v<Callable, Operation *>>
948 addDynamicallyLegalOp<OpT>(
949 [=](
Operation *op) {
return callback(cast<OpT>(op)); });
957 template <
typename OpT>
961 template <
typename OpT,
typename OpT2,
typename... OpTs>
974 template <
typename OpT>
979 template <
typename OpT,
typename OpT2,
typename... OpTs>
981 markOpRecursivelyLegal<OpT>(callback);
984 template <
typename OpT,
class Callable>
985 std::enable_if_t<!std::is_invocable_v<Callable, Operation *>>
987 markOpRecursivelyLegal<OpT>(
988 [=](
Operation *op) {
return callback(cast<OpT>(op)); });
996 template <
typename... Names>
1001 template <
typename... Args>
1009 template <
typename... Names>
1011 StringRef name, Names... names) {
1014 setLegalityCallback(dialectNames, callback);
1016 template <
typename... Args>
1019 Args::getDialectNamespace()...);
1026 setLegalityCallback(fn);
1031 template <
typename... Names>
1036 template <
typename... Args>
1078 struct LegalizationInfo {
1083 bool isRecursivelyLegal =
false;
1090 std::optional<LegalizationInfo> getOpInfo(OperationName op)
const;
1094 llvm::MapVector<OperationName, LegalizationInfo> legalOperations;
1098 DenseMap<OperationName, DynamicLegalityCallbackFn> opRecursiveLegalityFns;
1102 llvm::StringMap<LegalizationAction> legalDialects;
1105 llvm::StringMap<DynamicLegalityCallbackFn> dialectLegalityFns;
1114 #if MLIR_ENABLE_PDL_IN_PATTERNMATCH
1150 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 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 inlineBlockBefore(Block *source, Block *dest, Block::iterator before, ValueRange argValues={}) override
PatternRewriter hook for inlining the ops of a block into another block.
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, ValueRange to)
Replace all the uses of the block argument from with 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={})
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...