MLIR  16.0.0git
Static Public Member Functions | List of all members
mlir::OpTrait::AttrSizedOperandSegments< ConcreteType > Class Template Reference

A trait for operations that have an attribute specifying operand segments. More...

#include "mlir/IR/OpDefinition.h"

+ Inheritance diagram for mlir::OpTrait::AttrSizedOperandSegments< ConcreteType >:
+ Collaboration diagram for mlir::OpTrait::AttrSizedOperandSegments< ConcreteType >:

Static Public Member Functions

static StringRef getOperandSegmentSizeAttr ()
static LogicalResult verifyTrait (Operation *op)

Additional Inherited Members

- Protected Member Functions inherited from mlir::OpTrait::TraitBase< ConcreteType, AttrSizedOperandSegments >
OperationgetOperation ()
 Return the ultimate Operation being worked on. More...

Detailed Description

template<typename ConcreteType>
class mlir::OpTrait::AttrSizedOperandSegments< ConcreteType >

A trait for operations that have an attribute specifying operand segments.

Certain operations can have multiple variadic operands and their size relationship is not always known statically. For such cases, we need a per-op-instance specification to divide the operands into logical groups or segments. This can be modeled by attributes. The attribute will be named as operand_segment_sizes.

This trait verifies the attribute for specifying operand segments has the correct type (1D vector) and values (non-negative), etc.

Definition at line 1277 of file OpDefinition.h.

Member Function Documentation

◆ getOperandSegmentSizeAttr()

template<typename ConcreteType >
static StringRef mlir::OpTrait::AttrSizedOperandSegments< ConcreteType >::getOperandSegmentSizeAttr ( )

Definition at line 1280 of file OpDefinition.h.

Referenced by mlir::gpu::addAsyncDependency().

◆ verifyTrait()

template<typename ConcreteType >
static LogicalResult mlir::OpTrait::AttrSizedOperandSegments< ConcreteType >::verifyTrait ( Operation op)

Definition at line 1284 of file OpDefinition.h.

References mlir::OpTrait::impl::verifyOperandSizeAttr().

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