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

Public Member Functions

 AllReduceOpAdaptor (AllReduceOp op)
 
::llvm::LogicalResult verify (::mlir::Location loc)
 
 AllReduceOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions={})
 
 AllReduceOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs, ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions={})
 
 AllReduceOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs=nullptr)
 
template<typename LateInst = AllReduceOp, typename = std::enable_if_t<std::is_same_v<LateInst, AllReduceOp>>>
 AllReduceOpGenericAdaptor (RangeT values, LateInst op)
 
- Public Member Functions inherited from mlir::tt::ttir::AllReduceOpGenericAdaptor<::mlir::ValueRange >
 AllReduceOpGenericAdaptor (::mlir::ValueRange values, ::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions={})
 
 AllReduceOpGenericAdaptor (::mlir::ValueRange values, ::mlir::DictionaryAttr attrs, ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions={})
 
 AllReduceOpGenericAdaptor (::mlir::ValueRange values, ::mlir::DictionaryAttr attrs=nullptr)
 
 AllReduceOpGenericAdaptor (::mlir::ValueRange values, LateInst op)
 
std::pair< unsigned, unsigned > getODSOperandIndexAndLength (unsigned index)
 
::mlir::ValueRange getODSOperands (unsigned index)
 
::mlir::ValueRange getInputs ()
 
ValueT getOutput ()
 
::mlir::ValueRange getOperands ()
 
- Public Member Functions inherited from mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase
 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 ()
 

Additional Inherited Members

- Protected Attributes inherited from mlir::tt::ttir::detail::AllReduceOpGenericAdaptorBase
::mlir::DictionaryAttr odsAttrs
 
::std::optional<::mlir::OperationName > odsOpName
 
Properties properties
 
::mlir::RegionRange odsRegions
 

Constructor & Destructor Documentation

◆ AllReduceOpAdaptor()

mlir::tt::ttir::AllReduceOpAdaptor::AllReduceOpAdaptor ( AllReduceOp  op)

Member Function Documentation

◆ AllReduceOpGenericAdaptor() [1/4]

◆ AllReduceOpGenericAdaptor() [2/4]

◆ AllReduceOpGenericAdaptor() [3/4]

◆ AllReduceOpGenericAdaptor() [4/4]

template<typename LateInst = AllReduceOp, typename = std::enable_if_t<std::is_same_v<LateInst, AllReduceOp>>>
mlir::tt::ttir::AllReduceOpGenericAdaptor< RangeT >::AllReduceOpGenericAdaptor ( typename LateInst  = AllReduceOp,
typename  = std::enable_if_t<std::is_same_v<LateInst, AllReduceOp>> 
)
inline

◆ verify()

llvm::LogicalResult mlir::tt::ttir::AllReduceOpAdaptor::verify ( ::mlir::Location  loc)

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