mlir::mesh::IndependentParallelIteratorDomainShardingInterface< Op > Struct Template Reference

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

Inheritance diagram for mlir::mesh::IndependentParallelIteratorDomainShardingInterface< Op >:

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

Detailed Description

template<typename Op>
struct mlir::mesh::IndependentParallelIteratorDomainShardingInterface< Op >

Definition at line 56 of file ShardingInterfaceImpl.h.

Member Function Documentation

getIndexingMaps()

template<typename Op >

SmallVector<AffineMap> mlir::mesh::IndependentParallelIteratorDomainShardingInterface< Op >::getIndexingMaps ( Operation *  op ) const

inline

Definition at line 71 of file ShardingInterfaceImpl.h.

getLoopIteratorTypes()

template<typename Op >

SmallVector<utils::IteratorType> mlir::mesh::IndependentParallelIteratorDomainShardingInterface< Op >::getLoopIteratorTypes ( Operation *  operation ) const

inline

Definition at line 60 of file ShardingInterfaceImpl.h.

References mlir::Operation::getOperandTypes(), and mlir::Operation::getResultTypes().

spmdize()

template<typename Op >

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

inline

Definition at line 76 of file ShardingInterfaceImpl.h.

References mlir::mesh::spmdizeTriviallyShardableOperation().

---

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

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