mlir::tt::ttir::AllReduceOp Class Reference

Inheritance diagram for mlir::tt::ttir::AllReduceOp:

Public Types
using	Adaptor = AllReduceOpAdaptor
template<typename RangeT >
using	GenericAdaptor = AllReduceOpGenericAdaptor< RangeT >
using	FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute > >
using	Properties = FoldAdaptor::Properties

Public Member Functions
static ::llvm::ArrayRef<::llvm::StringRef >	getAttributeNames ()
::mlir::StringAttr	getChannelHandleAttrName ()
::mlir::StringAttr	getDimAttrName ()
::mlir::StringAttr	getReduceTypeAttrName ()
::mlir::StringAttr	getReplicaGroupsAttrName ()
::mlir::StringAttr	getUseGlobalDeviceIdsAttrName ()
std::pair< unsigned, unsigned >	getODSOperandIndexAndLength (unsigned index)
::mlir::Operation::operand_range	getODSOperands (unsigned index)
::mlir::Operation::operand_range	getInputs ()
::mlir::TypedValue<::mlir::RankedTensorType >	getOutput ()
::mlir::MutableOperandRange	getInputsMutable ()
::mlir::OpOperand &	getOutputMutable ()
std::pair< unsigned, unsigned >	getODSResultIndexAndLength (unsigned index)
::mlir::Operation::result_range	getODSResults (unsigned index)
::mlir::Operation::result_range	getResults ()
void	writeProperties (::mlir::DialectBytecodeWriter &writer)
::mlir::DenseIntElementsAttr	getReplicaGroupsAttr ()
::mlir::DenseIntElementsAttr	getReplicaGroups ()
::mlir::IntegerAttr	getDimAttr ()
int32_t	getDim ()
::mlir::IntegerAttr	getChannelHandleAttr ()
::std::optional< int32_t >	getChannelHandle ()
::mlir::UnitAttr	getUseGlobalDeviceIdsAttr ()
bool	getUseGlobalDeviceIds ()
::mlir::tt::ReduceTypeAttr	getReduceTypeAttr ()
::mlir::tt::ReduceType	getReduceType ()
void	setReplicaGroupsAttr (::mlir::DenseIntElementsAttr attr)
void	setDimAttr (::mlir::IntegerAttr attr)
void	setDim (int32_t attrValue)
void	setChannelHandleAttr (::mlir::IntegerAttr attr)
void	setChannelHandle (::std::optional< int32_t > attrValue)
void	setUseGlobalDeviceIdsAttr (::mlir::UnitAttr attr)
void	setUseGlobalDeviceIds (bool attrValue)
void	setReduceTypeAttr (::mlir::tt::ReduceTypeAttr attr)
void	setReduceType (::mlir::tt::ReduceType attrValue)
::mlir::Attribute	removeChannelHandleAttr ()
::mlir::Attribute	removeUseGlobalDeviceIdsAttr ()
::llvm::LogicalResult	verifyInvariantsImpl ()
::llvm::LogicalResult	verifyInvariants ()
::llvm::LogicalResult	verify ()
void	getEffects (::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> &effects)
MutableOperandRange	getDpsInitsMutable ()

Static Public Member Functions
::mlir::StringAttr	getChannelHandleAttrName (::mlir::OperationName name)
::mlir::StringAttr	getDimAttrName (::mlir::OperationName name)
::mlir::StringAttr	getReduceTypeAttrName (::mlir::OperationName name)
::mlir::StringAttr	getReplicaGroupsAttrName (::mlir::OperationName name)
::mlir::StringAttr	getUseGlobalDeviceIdsAttrName (::mlir::OperationName name)
static constexpr ::llvm::StringLiteral	getOperationName ()
::llvm::LogicalResult	setPropertiesFromAttr (Properties &prop, ::mlir::Attribute attr, ::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError)
::mlir::Attribute	getPropertiesAsAttr (::mlir::MLIRContext *ctx, const Properties &prop)
static llvm::hash_code	computePropertiesHash (const Properties &prop)
static std::optional< mlir::Attribute >	getInherentAttr (::mlir::MLIRContext *ctx, const Properties &prop, llvm::StringRef name)
static void	setInherentAttr (Properties &prop, llvm::StringRef name, mlir::Attribute value)
static void	populateInherentAttrs (::mlir::MLIRContext *ctx, const Properties &prop, ::mlir::NamedAttrList &attrs)
::llvm::LogicalResult	verifyInherentAttrs (::mlir::OperationName opName, ::mlir::NamedAttrList &attrs, llvm::function_ref<::mlir::InFlightDiagnostic()> emitError)
::llvm::LogicalResult	readProperties (::mlir::DialectBytecodeReader &reader, ::mlir::OperationState &state)
static void	build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange results, ::mlir::ValueRange inputs, ::mlir::Value output, ::mlir::DenseIntElementsAttr replica_groups, ::mlir::IntegerAttr dim, ::mlir::IntegerAttr channel_handle, ::mlir::UnitAttr use_global_device_ids, ::mlir::tt::ReduceTypeAttr reduce_type)
static void	build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange results, ::mlir::ValueRange inputs, ::mlir::Value output, ::mlir::DenseIntElementsAttr replica_groups, int32_t dim, ::mlir::IntegerAttr channel_handle, bool use_global_device_ids, ::mlir::tt::ReduceType reduce_type)
static void	build (::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute > attributes={})

Private Member Functions
::mlir::StringAttr	getAttributeNameForIndex (unsigned index)

Static Private Member Functions
::mlir::StringAttr	getAttributeNameForIndex (::mlir::OperationName name, unsigned index)

Member Typedef Documentation

Adaptor

using mlir::tt::ttir::AllReduceOp::Adaptor = AllReduceOpAdaptor

FoldAdaptor

using mlir::tt::ttir::AllReduceOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> >

GenericAdaptor

template<typename RangeT >

using mlir::tt::ttir::AllReduceOp::GenericAdaptor = AllReduceOpGenericAdaptor<RangeT>

Properties

using mlir::tt::ttir::AllReduceOp::Properties = FoldAdaptor::Properties

Member Function Documentation

build() [1/3]

void mlir::tt::ttir::AllReduceOp::build ( ::mlir::OpBuilder &  , ::mlir::OperationState &  odsState, ::mlir::TypeRange  resultTypes, ::mlir::ValueRange  operands, ::llvm::ArrayRef<::mlir::NamedAttribute >  attributes = {}  )

static

build() [2/3]

void mlir::tt::ttir::AllReduceOp::build ( ::mlir::OpBuilder &  odsBuilder, ::mlir::OperationState &  odsState, ::mlir::TypeRange  results, ::mlir::ValueRange  inputs, ::mlir::Value  output, ::mlir::DenseIntElementsAttr  replica_groups, ::mlir::IntegerAttr  dim, ::mlir::IntegerAttr  channel_handle, ::mlir::UnitAttr  use_global_device_ids, ::mlir::tt::ReduceTypeAttr  reduce_type  )

static

build() [3/3]

void mlir::tt::ttir::AllReduceOp::build ( ::mlir::OpBuilder &  odsBuilder, ::mlir::OperationState &  odsState, ::mlir::TypeRange  results, ::mlir::ValueRange  inputs, ::mlir::Value  output, ::mlir::DenseIntElementsAttr  replica_groups, int32_t  dim, ::mlir::IntegerAttr  channel_handle, bool  use_global_device_ids, ::mlir::tt::ReduceType  reduce_type  )

static

computePropertiesHash()

llvm::hash_code mlir::tt::ttir::AllReduceOp::computePropertiesHash ( const Properties &  prop )

static

getAttributeNameForIndex() [1/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getAttributeNameForIndex ( ::mlir::OperationName  name, unsigned  index  )

inlinestaticprivate

getAttributeNameForIndex() [2/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getAttributeNameForIndex ( unsigned  index )

inlineprivate

getAttributeNames()

static ::llvm::ArrayRef<::llvm::StringRef> mlir::tt::ttir::AllReduceOp::getAttributeNames ( )

inline

getChannelHandle()

std::optional< int32_t > mlir::tt::ttir::AllReduceOp::getChannelHandle

(

)

getChannelHandleAttr()

::mlir::IntegerAttr mlir::tt::ttir::AllReduceOp::getChannelHandleAttr ( )

inline

getChannelHandleAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getChannelHandleAttrName ( )

inline

getChannelHandleAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getChannelHandleAttrName ( ::mlir::OperationName  name )

inlinestatic

getDim()

int32_t mlir::tt::ttir::AllReduceOp::getDim

(

)

getDimAttr()

::mlir::IntegerAttr mlir::tt::ttir::AllReduceOp::getDimAttr ( )

inline

getDimAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getDimAttrName ( )

inline

getDimAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getDimAttrName ( ::mlir::OperationName  name )

inlinestatic

getDpsInitsMutable()

MutableOperandRange mlir::tt::ttir::AllReduceOp::getDpsInitsMutable ( )

inline

getEffects()

void mlir::tt::ttir::AllReduceOp::getEffects

(

::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> &

effects

)

getInherentAttr()

std::optional< mlir::Attribute > mlir::tt::ttir::AllReduceOp::getInherentAttr ( ::mlir::MLIRContext *  ctx, const Properties &  prop, llvm::StringRef  name  )

static

getInputs()

::mlir::Operation::operand_range mlir::tt::ttir::AllReduceOp::getInputs ( )

inline

getInputsMutable()

mlir::MutableOperandRange mlir::tt::ttir::AllReduceOp::getInputsMutable

(

)

getODSOperandIndexAndLength()

std::pair< unsigned, unsigned > mlir::tt::ttir::AllReduceOp::getODSOperandIndexAndLength

(

unsigned

index

)

getODSOperands()

::mlir::Operation::operand_range mlir::tt::ttir::AllReduceOp::getODSOperands ( unsigned  index )

inline

getODSResultIndexAndLength()

std::pair< unsigned, unsigned > mlir::tt::ttir::AllReduceOp::getODSResultIndexAndLength

(

unsigned

index

)

getODSResults()

::mlir::Operation::result_range mlir::tt::ttir::AllReduceOp::getODSResults ( unsigned  index )

inline

getOperationName()

static constexpr ::llvm::StringLiteral mlir::tt::ttir::AllReduceOp::getOperationName ( )

inlinestatic

getOutput()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::AllReduceOp::getOutput ( )

inline

getOutputMutable()

::mlir::OpOperand& mlir::tt::ttir::AllReduceOp::getOutputMutable ( )

inline

getPropertiesAsAttr()

mlir::Attribute mlir::tt::ttir::AllReduceOp::getPropertiesAsAttr ( ::mlir::MLIRContext *  ctx, const Properties &  prop  )

static

getReduceType()

mlir::tt::ReduceType mlir::tt::ttir::AllReduceOp::getReduceType

(

)

getReduceTypeAttr()

::mlir::tt::ReduceTypeAttr mlir::tt::ttir::AllReduceOp::getReduceTypeAttr ( )

inline

getReduceTypeAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getReduceTypeAttrName ( )

inline

getReduceTypeAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getReduceTypeAttrName ( ::mlir::OperationName  name )

inlinestatic

getReplicaGroups()

mlir::DenseIntElementsAttr mlir::tt::ttir::AllReduceOp::getReplicaGroups

(

)

getReplicaGroupsAttr()

::mlir::DenseIntElementsAttr mlir::tt::ttir::AllReduceOp::getReplicaGroupsAttr ( )

inline

getReplicaGroupsAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getReplicaGroupsAttrName ( )

inline

getReplicaGroupsAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getReplicaGroupsAttrName ( ::mlir::OperationName  name )

inlinestatic

getResults()

::mlir::Operation::result_range mlir::tt::ttir::AllReduceOp::getResults ( )

inline

getUseGlobalDeviceIds()

bool mlir::tt::ttir::AllReduceOp::getUseGlobalDeviceIds

(

)

getUseGlobalDeviceIdsAttr()

::mlir::UnitAttr mlir::tt::ttir::AllReduceOp::getUseGlobalDeviceIdsAttr ( )

inline

getUseGlobalDeviceIdsAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getUseGlobalDeviceIdsAttrName ( )

inline

getUseGlobalDeviceIdsAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttir::AllReduceOp::getUseGlobalDeviceIdsAttrName ( ::mlir::OperationName  name )

inlinestatic

populateInherentAttrs()

void mlir::tt::ttir::AllReduceOp::populateInherentAttrs ( ::mlir::MLIRContext *  ctx, const Properties &  prop, ::mlir::NamedAttrList &  attrs  )

static

readProperties()

llvm::LogicalResult mlir::tt::ttir::AllReduceOp::readProperties ( ::mlir::DialectBytecodeReader &  reader, ::mlir::OperationState &  state  )

static

removeChannelHandleAttr()

::mlir::Attribute mlir::tt::ttir::AllReduceOp::removeChannelHandleAttr ( )

inline

removeUseGlobalDeviceIdsAttr()

::mlir::Attribute mlir::tt::ttir::AllReduceOp::removeUseGlobalDeviceIdsAttr ( )

inline

setChannelHandle()

void mlir::tt::ttir::AllReduceOp::setChannelHandle

(

::std::optional< int32_t >

attrValue

)

setChannelHandleAttr()

void mlir::tt::ttir::AllReduceOp::setChannelHandleAttr ( ::mlir::IntegerAttr  attr )

inline

setDim()

void mlir::tt::ttir::AllReduceOp::setDim

(

int32_t

attrValue

)

setDimAttr()

void mlir::tt::ttir::AllReduceOp::setDimAttr ( ::mlir::IntegerAttr  attr )

inline

setInherentAttr()

void mlir::tt::ttir::AllReduceOp::setInherentAttr ( Properties &  prop, llvm::StringRef  name, mlir::Attribute  value  )

static

setPropertiesFromAttr()

llvm::LogicalResult mlir::tt::ttir::AllReduceOp::setPropertiesFromAttr ( Properties &  prop, ::mlir::Attribute  attr, ::llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError  )

static

setReduceType()

void mlir::tt::ttir::AllReduceOp::setReduceType

(

::mlir::tt::ReduceType

attrValue

)

setReduceTypeAttr()

void mlir::tt::ttir::AllReduceOp::setReduceTypeAttr ( ::mlir::tt::ReduceTypeAttr  attr )

inline

setReplicaGroupsAttr()

void mlir::tt::ttir::AllReduceOp::setReplicaGroupsAttr ( ::mlir::DenseIntElementsAttr  attr )

inline

setUseGlobalDeviceIds()

void mlir::tt::ttir::AllReduceOp::setUseGlobalDeviceIds

(

bool

attrValue

)

setUseGlobalDeviceIdsAttr()

void mlir::tt::ttir::AllReduceOp::setUseGlobalDeviceIdsAttr ( ::mlir::UnitAttr  attr )

inline

verify()

::llvm::LogicalResult mlir::tt::ttir::AllReduceOp::verify

(

)

verifyInherentAttrs()

llvm::LogicalResult mlir::tt::ttir::AllReduceOp::verifyInherentAttrs ( ::mlir::OperationName  opName, ::mlir::NamedAttrList &  attrs, llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError  )

static

verifyInvariants()

llvm::LogicalResult mlir::tt::ttir::AllReduceOp::verifyInvariants

(

)

verifyInvariantsImpl()

llvm::LogicalResult mlir::tt::ttir::AllReduceOp::verifyInvariantsImpl

(

)

writeProperties()

void mlir::tt::ttir::AllReduceOp::writeProperties

(

::mlir::DialectBytecodeWriter &

writer

)

---

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

• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTIR/IR/TTIROps.h.inc
• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTIR/IR/TTIROps.cpp.inc