MLIR 23.0.0git
mlir::DynamicOpDefinition Class Reference

The definition of a dynamic op. More...

#include "mlir/IR/ExtensibleDialect.h"

Inheritance diagram for mlir::DynamicOpDefinition:

Public Types

using GetCanonicalizationPatternsFn

Public Member Functions

TypeID getTypeID ()
 Returns the op typeID.
void setVerifyFn (OperationName::VerifyInvariantsFn &&verify)
 Sets the verifier function for this operation.
void setVerifyRegionFn (OperationName::VerifyRegionInvariantsFn &&verify)
 Sets the region verifier function for this operation.
void setParseFn (OperationName::ParseAssemblyFn &&parse)
 Sets the static hook for parsing this op assembly.
void setPrintFn (OperationName::PrintAssemblyFn &&print)
 Sets the static hook for printing this op assembly.
void setFoldHookFn (OperationName::FoldHookFn &&foldHook)
 Sets the hook implementing a generalized folder for the op.
void setGetCanonicalizationPatternsFn (GetCanonicalizationPatternsFn &&getCanonicalizationPatterns)
 Set the hook returning any canonicalization pattern rewrites that the op supports, for use by the canonicalization pass.
void setPopulateDefaultAttrsFn (OperationName::PopulateDefaultAttrsFn &&populateDefaultAttrs)
 Set the hook populating default attributes.
bool addTrait (std::unique_ptr< DynamicOpTrait > trait)
 Attach a trait to this dynamic-defined op.
LogicalResult foldHook (Operation *op, ArrayRef< Attribute > attrs, SmallVectorImpl< OpFoldResult > &results) final
void getCanonicalizationPatterns (RewritePatternSet &set, MLIRContext *context) final
bool hasTrait (TypeID id) final
OperationName::ParseAssemblyFn getParseAssemblyFn () final
void populateDefaultAttrs (const OperationName &name, NamedAttrList &attrs) final
void printAssembly (Operation *op, OpAsmPrinter &printer, StringRef name) final
LogicalResult verifyInvariants (Operation *op) final
LogicalResult verifyRegionInvariants (Operation *op) final
std::optional< AttributegetInherentAttr (Operation *op, StringRef name) final
 Implementation for properties (unsupported right now here).
void setInherentAttr (Operation *op, StringAttr name, Attribute value) final
void populateInherentAttrs (Operation *op, NamedAttrList &attrs) final
LogicalResult verifyInherentAttrs (OperationName opName, NamedAttrList &attributes, function_ref< InFlightDiagnostic()> emitError) final
int getOpPropertyByteSize () final
void initProperties (OperationName opName, OpaqueProperties storage, OpaqueProperties init) final
void deleteProperties (OpaqueProperties prop) final
void populateDefaultProperties (OperationName opName, OpaqueProperties properties) final
LogicalResult setPropertiesFromAttr (OperationName opName, OpaqueProperties properties, Attribute attr, function_ref< InFlightDiagnostic()> emitError) final
Attribute getPropertiesAsAttr (Operation *op) final
void copyProperties (OpaqueProperties lhs, OpaqueProperties rhs) final
bool compareProperties (OpaqueProperties, OpaqueProperties) final
llvm::hash_code hashProperties (OpaqueProperties prop) final
Public Member Functions inherited from mlir::OperationName::Impl
 Impl (StringRef, Dialect *dialect, TypeID typeID, detail::InterfaceMap interfaceMap)
 Impl (StringAttr name, Dialect *dialect, TypeID typeID, detail::InterfaceMap interfaceMap)
bool isRegistered () const
 Returns true if this is a registered operation.
detail::InterfaceMapgetInterfaceMap ()
DialectgetDialect () const
StringAttr getName () const
TypeID getTypeID () const
ArrayRef< StringAttr > getAttributeNames () const
Public Member Functions inherited from mlir::OperationName::InterfaceConcept
virtual ~InterfaceConcept ()=default

Static Public Member Functions

static std::unique_ptr< DynamicOpDefinitionget (StringRef name, ExtensibleDialect *dialect, OperationName::VerifyInvariantsFn &&verifyFn, OperationName::VerifyRegionInvariantsFn &&verifyRegionFn)
 Create a new op at runtime.
static std::unique_ptr< DynamicOpDefinitionget (StringRef name, ExtensibleDialect *dialect, OperationName::VerifyInvariantsFn &&verifyFn, OperationName::VerifyRegionInvariantsFn &&verifyRegionFn, OperationName::ParseAssemblyFn &&parseFn, OperationName::PrintAssemblyFn &&printFn)
static std::unique_ptr< DynamicOpDefinitionget (StringRef name, ExtensibleDialect *dialect, OperationName::VerifyInvariantsFn &&verifyFn, OperationName::VerifyRegionInvariantsFn &&verifyRegionFn, OperationName::ParseAssemblyFn &&parseFn, OperationName::PrintAssemblyFn &&printFn, OperationName::FoldHookFn &&foldHookFn, GetCanonicalizationPatternsFn &&getCanonicalizationPatternsFn, OperationName::PopulateDefaultAttrsFn &&populateDefaultAttrsFn)

Additional Inherited Members

Protected Attributes inherited from mlir::OperationName::Impl
StringAttr name
 The name of the operation.
TypeID typeID
 The unique identifier of the derived Op class.
Dialectdialect
 The following fields are only populated when the operation is registered.
detail::InterfaceMap interfaceMap
 A map of interfaces that were registered to this operation.
ArrayRef< StringAttr > attributeNames
 A list of attribute names registered to this operation in StringAttr form.

Detailed Description

The definition of a dynamic op.

A dynamic op is an op that is defined at runtime, and that can be registered at runtime by an extensible dialect (a dialect inheriting ExtensibleDialect). This class implements the method exposed by the OperationName class, and in addition defines the TypeID of the op that will be defined. Each dynamic operation definition refers to one instance of this class.

Definition at line 431 of file ExtensibleDialect.h.

Member Typedef Documentation

◆ GetCanonicalizationPatternsFn

Initial value:
llvm::unique_function<void(RewritePatternSet &, MLIRContext *) const>
MLIRContext is the top-level object for a collection of MLIR operations.
Definition MLIRContext.h:63

Definition at line 433 of file ExtensibleDialect.h.

Member Function Documentation

◆ addTrait()

bool mlir::DynamicOpDefinition::addTrait ( std::unique_ptr< DynamicOpTrait > trait)
inline

Attach a trait to this dynamic-defined op.

Definition at line 505 of file ExtensibleDialect.h.

◆ compareProperties()

bool mlir::DynamicOpDefinition::compareProperties ( OpaqueProperties ,
OpaqueProperties  )
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 569 of file ExtensibleDialect.h.

◆ copyProperties()

void mlir::DynamicOpDefinition::copyProperties ( OpaqueProperties lhs,
OpaqueProperties rhs )
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 568 of file ExtensibleDialect.h.

References lhs, and rhs.

◆ deleteProperties()

void mlir::DynamicOpDefinition::deleteProperties ( OpaqueProperties prop)
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 556 of file ExtensibleDialect.h.

◆ foldHook()

LogicalResult mlir::DynamicOpDefinition::foldHook ( Operation * op,
ArrayRef< Attribute > attrs,
SmallVectorImpl< OpFoldResult > & results )
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 509 of file ExtensibleDialect.h.

Referenced by setFoldHookFn().

◆ get() [1/3]

std::unique_ptr< DynamicOpDefinition > DynamicOpDefinition::get ( StringRef name,
ExtensibleDialect * dialect,
OperationName::VerifyInvariantsFn && verifyFn,
OperationName::VerifyRegionInvariantsFn && verifyRegionFn )
static

Create a new op at runtime.

The op is registered only after passing it to the dialect using registerDynamicOp.

Definition at line 309 of file ExtensibleDialect.cpp.

References mlir::OperationName::Impl::dialect, mlir::AsmParser::emitError(), get(), mlir::AsmParser::getCurrentLocation(), mlir::OperationName::Impl::name, and result.

Referenced by get(), get(), and loadOperation().

◆ get() [2/3]

std::unique_ptr< DynamicOpDefinition > DynamicOpDefinition::get ( StringRef name,
ExtensibleDialect * dialect,
OperationName::VerifyInvariantsFn && verifyFn,
OperationName::VerifyRegionInvariantsFn && verifyRegionFn,
OperationName::ParseAssemblyFn && parseFn,
OperationName::PrintAssemblyFn && printFn )
static

◆ get() [3/3]

std::unique_ptr< DynamicOpDefinition > DynamicOpDefinition::get ( StringRef name,
ExtensibleDialect * dialect,
OperationName::VerifyInvariantsFn && verifyFn,
OperationName::VerifyRegionInvariantsFn && verifyRegionFn,
OperationName::ParseAssemblyFn && parseFn,
OperationName::PrintAssemblyFn && printFn,
OperationName::FoldHookFn && foldHookFn,
GetCanonicalizationPatternsFn && getCanonicalizationPatternsFn,
OperationName::PopulateDefaultAttrsFn && populateDefaultAttrsFn )
static

◆ getCanonicalizationPatterns()

void mlir::DynamicOpDefinition::getCanonicalizationPatterns ( RewritePatternSet & set,
MLIRContext * context )
inlinefinalvirtual

◆ getInherentAttr()

std::optional< Attribute > mlir::DynamicOpDefinition::getInherentAttr ( Operation * op,
StringRef name )
inlinefinalvirtual

Implementation for properties (unsupported right now here).

Implements mlir::OperationName::InterfaceConcept.

Definition at line 540 of file ExtensibleDialect.h.

References mlir::OperationName::Impl::name.

◆ getOpPropertyByteSize()

int mlir::DynamicOpDefinition::getOpPropertyByteSize ( )
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 553 of file ExtensibleDialect.h.

◆ getParseAssemblyFn()

OperationName::ParseAssemblyFn mlir::DynamicOpDefinition::getParseAssemblyFn ( )
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 518 of file ExtensibleDialect.h.

◆ getPropertiesAsAttr()

Attribute mlir::DynamicOpDefinition::getPropertiesAsAttr ( Operation * op)
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 567 of file ExtensibleDialect.h.

◆ getTypeID()

TypeID mlir::DynamicOpDefinition::getTypeID ( )
inline

Returns the op typeID.

Definition at line 459 of file ExtensibleDialect.h.

References mlir::OperationName::Impl::typeID.

◆ hashProperties()

llvm::hash_code mlir::DynamicOpDefinition::hashProperties ( OpaqueProperties prop)
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 572 of file ExtensibleDialect.h.

◆ hasTrait()

bool mlir::DynamicOpDefinition::hasTrait ( TypeID id)
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 517 of file ExtensibleDialect.h.

◆ initProperties()

void mlir::DynamicOpDefinition::initProperties ( OperationName opName,
OpaqueProperties storage,
OpaqueProperties init )
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 554 of file ExtensibleDialect.h.

◆ populateDefaultAttrs()

void mlir::DynamicOpDefinition::populateDefaultAttrs ( const OperationName & name,
NamedAttrList & attrs )
inlinefinalvirtual

◆ populateDefaultProperties()

void mlir::DynamicOpDefinition::populateDefaultProperties ( OperationName opName,
OpaqueProperties properties )
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 557 of file ExtensibleDialect.h.

◆ populateInherentAttrs()

void mlir::DynamicOpDefinition::populateInherentAttrs ( Operation * op,
NamedAttrList & attrs )
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 547 of file ExtensibleDialect.h.

◆ printAssembly()

void mlir::DynamicOpDefinition::printAssembly ( Operation * op,
OpAsmPrinter & printer,
StringRef name )
inlinefinalvirtual

◆ setFoldHookFn()

void mlir::DynamicOpDefinition::setFoldHookFn ( OperationName::FoldHookFn && foldHook)
inline

Sets the hook implementing a generalized folder for the op.

See RegisteredOperationName::foldHook for more details

Definition at line 487 of file ExtensibleDialect.h.

References foldHook().

◆ setGetCanonicalizationPatternsFn()

void mlir::DynamicOpDefinition::setGetCanonicalizationPatternsFn ( GetCanonicalizationPatternsFn && getCanonicalizationPatterns)
inline

Set the hook returning any canonicalization pattern rewrites that the op supports, for use by the canonicalization pass.

Definition at line 493 of file ExtensibleDialect.h.

References getCanonicalizationPatterns().

◆ setInherentAttr()

void mlir::DynamicOpDefinition::setInherentAttr ( Operation * op,
StringAttr name,
Attribute value )
inlinefinalvirtual

◆ setParseFn()

void mlir::DynamicOpDefinition::setParseFn ( OperationName::ParseAssemblyFn && parse)
inline

Sets the static hook for parsing this op assembly.

Definition at line 476 of file ExtensibleDialect.h.

◆ setPopulateDefaultAttrsFn()

void mlir::DynamicOpDefinition::setPopulateDefaultAttrsFn ( OperationName::PopulateDefaultAttrsFn && populateDefaultAttrs)
inline

Set the hook populating default attributes.

Definition at line 499 of file ExtensibleDialect.h.

References populateDefaultAttrs().

◆ setPrintFn()

void mlir::DynamicOpDefinition::setPrintFn ( OperationName::PrintAssemblyFn && print)
inline

Sets the static hook for printing this op assembly.

Definition at line 481 of file ExtensibleDialect.h.

References print().

◆ setPropertiesFromAttr()

LogicalResult mlir::DynamicOpDefinition::setPropertiesFromAttr ( OperationName opName,
OpaqueProperties properties,
Attribute attr,
function_ref< InFlightDiagnostic()> emitError )
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 561 of file ExtensibleDialect.h.

References mlir::emitError().

◆ setVerifyFn()

void mlir::DynamicOpDefinition::setVerifyFn ( OperationName::VerifyInvariantsFn && verify)
inline

Sets the verifier function for this operation.

It should emits an error message and returns failure if a problem is detected, or returns success if everything is ok.

Definition at line 464 of file ExtensibleDialect.h.

References mlir::verify().

◆ setVerifyRegionFn()

void mlir::DynamicOpDefinition::setVerifyRegionFn ( OperationName::VerifyRegionInvariantsFn && verify)
inline

Sets the region verifier function for this operation.

It should emits an error message and returns failure if a problem is detected, or returns success if everything is ok.

Definition at line 471 of file ExtensibleDialect.h.

References mlir::verify().

◆ verifyInherentAttrs()

LogicalResult mlir::DynamicOpDefinition::verifyInherentAttrs ( OperationName opName,
NamedAttrList & attributes,
function_ref< InFlightDiagnostic()> emitError )
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 549 of file ExtensibleDialect.h.

References mlir::emitError(), and success().

◆ verifyInvariants()

LogicalResult mlir::DynamicOpDefinition::verifyInvariants ( Operation * op)
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 531 of file ExtensibleDialect.h.

◆ verifyRegionInvariants()

LogicalResult mlir::DynamicOpDefinition::verifyRegionInvariants ( Operation * op)
inlinefinalvirtual

Implements mlir::OperationName::InterfaceConcept.

Definition at line 534 of file ExtensibleDialect.h.


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