TT-MLIR
Classes | Public Member Functions | Protected Attributes | List of all members
mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase Class Reference
Inheritance diagram for mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase:
mlir::tt::ttir::AllReduceOpGenericAdaptor<::mlir::ValueRange > mlir::tt::ttir::AllReduceOpGenericAdaptor< RangeT > mlir::tt::ttir::AllReduceOpAdaptor

Classes

struct  Properties
 

Public Member Functions

 AllReduceOpGenericAdaptorBase (::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions={})
 
 AllReduceOpGenericAdaptorBase (AllReduceOp op)
 
std::pair< unsigned, unsigned > getODSOperandIndexAndLength (unsigned index, unsigned odsOperandsSize)
 
const PropertiesgetProperties ()
 
::mlir::DictionaryAttr getAttributes ()
 
::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 ()
 

Protected Attributes

::mlir::DictionaryAttr odsAttrs
 
::std::optional<::mlir::OperationName > odsOpName
 
Properties properties
 
::mlir::RegionRange odsRegions
 

Constructor & Destructor Documentation

◆ AllReduceOpGenericAdaptorBase() [1/2]

mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::AllReduceOpGenericAdaptorBase ( ::mlir::DictionaryAttr  attrs,
const Properties properties,
::mlir::RegionRange  regions = {} 
)
inline

◆ AllReduceOpGenericAdaptorBase() [2/2]

mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::AllReduceOpGenericAdaptorBase ( AllReduceOp  op)

Member Function Documentation

◆ getAttributes()

::mlir::DictionaryAttr mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getAttributes ( )
inline

◆ getChannelHandle()

std::optional< int32_t > mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getChannelHandle ( )

◆ getChannelHandleAttr()

::mlir::IntegerAttr mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getChannelHandleAttr ( )
inline

◆ getDim()

int32_t mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getDim ( )

◆ getDimAttr()

::mlir::IntegerAttr mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getDimAttr ( )
inline

◆ getODSOperandIndexAndLength()

std::pair< unsigned, unsigned > mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getODSOperandIndexAndLength ( unsigned  index,
unsigned  odsOperandsSize 
)

◆ getProperties()

const Properties& mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getProperties ( )
inline

◆ getReduceType()

mlir::tt::ReduceType mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getReduceType ( )

◆ getReduceTypeAttr()

::mlir::tt::ReduceTypeAttr mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getReduceTypeAttr ( )
inline

◆ getReplicaGroups()

mlir::DenseIntElementsAttr mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getReplicaGroups ( )

◆ getReplicaGroupsAttr()

::mlir::DenseIntElementsAttr mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getReplicaGroupsAttr ( )
inline

◆ getUseGlobalDeviceIds()

bool mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getUseGlobalDeviceIds ( )

◆ getUseGlobalDeviceIdsAttr()

mlir::UnitAttr mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::getUseGlobalDeviceIdsAttr ( )

Member Data Documentation

◆ odsAttrs

::mlir::DictionaryAttr mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::odsAttrs
protected

◆ odsOpName

::std::optional<::mlir::OperationName> mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::odsOpName
protected

◆ odsRegions

::mlir::RegionRange mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::odsRegions
protected

◆ properties

Properties mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase::properties
protected

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