#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.
|
FailureOr< std::pair< bool, MeshShardingAttr > > | mlir::mesh::getMeshShardingAttr (OpResult result) |
|
FailureOr< std::pair< bool, MeshShardingAttr > > | mlir::mesh::getMeshShardingAttr (OpOperand &opOperand) |
|
FailureOr< ShardingOption > | mlir::mesh::detail::defaultGetShardingOption (Operation *op, ArrayRef< MeshShardingAttr > operandShardings, ArrayRef< MeshShardingAttr > resultShardings) |
|
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) |
|