MLIR  19.0.0git
Public Member Functions | List of all members
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< AffineMapgetIndexingMaps (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 Op>
struct mlir::mesh::IndependentParallelIteratorDomainShardingInterface< Op >

Definition at line 54 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 69 of file ShardingInterfaceImpl.h.

◆ getLoopIteratorTypes()

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

◆ spmdize()

template<typename Op >
LogicalResult mlir::mesh::IndependentParallelIteratorDomainShardingInterface< Op >::spmdize ( Operation op,
ArrayRef< Value spmdizedOperands,
ArrayRef< MeshShardingAttr >  operandShardings,
ArrayRef< MeshShardingAttr >  resultShardings,
IRMapping spmdizationMap,
SymbolTableCollection symbolTable,
OpBuilder builder 
) const
inline

Definition at line 74 of file ShardingInterfaceImpl.h.


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