mlir::mesh::ElementwiseShardingInterface< ElemwiseOp > Struct Template Reference

#include "mlir/Dialect/Mesh/Interfaces/ShardingInterfaceImpl.h"

Inheritance diagram for mlir::mesh::ElementwiseShardingInterface< ElemwiseOp >:

Public Member Functions
SmallVector< utils::IteratorType >	getLoopIteratorTypes (Operation *op) const
SmallVector< AffineMap >	getIndexingMaps (Operation *op) const
LogicalResult	spmdize (Operation *op, ArrayRef< Value > spmdizedOperands, ArrayRef< MeshShardingAttr > operandShardings, ArrayRef< MeshShardingAttr > resultShardings, IRMapping &spmdizationMap, SymbolTableCollection &symbolTable, OpBuilder &builder) const

Detailed Description

template<typename ElemwiseOp>
struct mlir::mesh::ElementwiseShardingInterface< ElemwiseOp >

Definition at line 104 of file ShardingInterfaceImpl.h.

Member Function Documentation

getIndexingMaps()

template<typename ElemwiseOp >

SmallVector<AffineMap> mlir::mesh::ElementwiseShardingInterface< ElemwiseOp >::getIndexingMaps ( Operation *  op ) const

inline

Definition at line 117 of file ShardingInterfaceImpl.h.

getLoopIteratorTypes()

template<typename ElemwiseOp >

SmallVector<utils::IteratorType> mlir::mesh::ElementwiseShardingInterface< ElemwiseOp >::getLoopIteratorTypes ( Operation *  op ) const

inline

Definition at line 107 of file ShardingInterfaceImpl.h.

spmdize()

template<typename ElemwiseOp >

LogicalResult mlir::mesh::ElementwiseShardingInterface< ElemwiseOp >::spmdize ( Operation *  op, ArrayRef< Value >  spmdizedOperands, ArrayRef< MeshShardingAttr >  operandShardings, ArrayRef< MeshShardingAttr >  resultShardings, IRMapping &  spmdizationMap, SymbolTableCollection &  symbolTable, OpBuilder &  builder  ) const

inline

Definition at line 130 of file ShardingInterfaceImpl.h.

---

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

• include/mlir/Dialect/Mesh/Interfaces/ShardingInterfaceImpl.h