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, MeshSharding > >	mlir::mesh::getMeshSharding (OpResult result)
FailureOr< std::pair< bool, MeshSharding > >	mlir::mesh::getMeshSharding (OpOperand &opOperand)
FailureOr< ShardingOption >	mlir::mesh::detail::defaultGetShardingOption (Operation *op, ArrayRef< MeshSharding > operandShardings, ArrayRef< MeshSharding > resultShardings)
FailureOr< std::vector< MeshSharding > >	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< MeshSharding > operandShardings, ArrayRef< MeshSharding > resultShardings, IRMapping &spmdizationMap, SymbolTableCollection &symbolTable, OpBuilder &builder)