MLIR  19.0.0git
Classes | Namespaces | Typedefs | Functions
ShardingInterface.h File Reference
#include "mlir/Dialect/Mesh/IR/MeshOps.h"
#include "mlir/Dialect/Utils/StructuredOpsUtils.h"
#include "mlir/IR/Value.h"
#include "mlir/Support/LLVM.h"
#include "mlir/Dialect/Mesh/Interfaces/ShardingInterface.h.inc"

Go to the source code of this file.

Classes

struct  mlir::mesh::ShardingOption
 

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::mesh
 
 mlir::mesh::detail
 

Typedefs

using mlir::mesh::ShardingArray = SmallVector< SmallVector< MeshAxis > >
 
using mlir::mesh::ShardingArrayRef = ArrayRef< SmallVector< MeshAxis > >
 

Functions

FailureOr< std::pair< bool, MeshShardingAttr > > mlir::mesh::getMeshShardingAttr (OpResult result)
 
FailureOr< std::pair< bool, MeshShardingAttr > > mlir::mesh::getMeshShardingAttr (OpOperand &opOperand)
 
FailureOr< ShardingOptionmlir::mesh::detail::defaultGetShardingOption (Operation *op, ArrayRef< MeshShardingAttr > operandShardings, ArrayRef< MeshShardingAttr > resultShardings)
 
FailureOr< SmallVector< MeshShardingAttr > > mlir::mesh::detail::defaultGetShardingAnnotations (Operation *op, const ShardingOption &shardingOption)
 
LogicalResult mlir::mesh::detail::defaultAddShardingAnnotations (Operation *op, OpBuilder &b, const ShardingOption &shardingOption)
 
void mlir::mesh::spmdizeFullyReplicatedOperation (Operation &op, ArrayRef< Value > spmdizedOperands, ArrayRef< MeshShardingAttr > operandShardings, ArrayRef< MeshShardingAttr > resultShardings, IRMapping &spmdizationMap, SymbolTableCollection &symbolTable, OpBuilder &builder)