|
| bool | mlir::shard::isReductionLoop (utils::IteratorType iType) |
| |
| template<typename T > |
| void | mlir::shard::removeTrailingEmptySubArray (SmallVector< SmallVector< T >> &array) |
| |
| bool | mlir::shard::isFullReplication (Sharding sharding) |
| |
| shard::GridOp | mlir::shard::getGridOrNull (Operation *op, FlatSymbolRefAttr gridSymbol, SymbolTableCollection &symbolTableCollection) |
| |
| shard::GridOp | mlir::shard::getGrid (Operation *op, FlatSymbolRefAttr gridSymbol, SymbolTableCollection &symbolTableCollection) |
| |
| template<typename Op > |
| shard::GridOp | mlir::shard::getGrid (Op op, SymbolTableCollection &symbolTableCollection) |
| |
| template<> |
| shard::GridOp | mlir::shard::getGrid< ShardOp > (ShardOp op, SymbolTableCollection &symbolTableCollection) |
| |
| template<typename GridAxesRange , typename GridShapeRange > |
| int64_t | mlir::shard::collectiveProcessGroupSize (GridAxesRange &&gridAxes, GridShapeRange &&gridShape) |
| |
| template<typename GridAxesRange > |
| int64_t | mlir::shard::collectiveProcessGroupSize (GridAxesRange &&gridAxes, GridOp grid) |
| |
| int64_t | mlir::shard::shardDimension (int64_t dimSize, int64_t shardCount) |
| |
| int64_t | mlir::shard::gatherDimension (int64_t dimSize, int64_t shardCount) |
| |
| ShapedType | mlir::shard::shardShapedType (ShapedType shape, GridOp grid, Sharding sharding) |
| |
| Type | mlir::shard::shardType (Type type, GridOp grid, Sharding sharding) |
| |
| void | mlir::shard::maybeInsertTargetShardingAnnotation (Sharding sharding, OpResult result, OpBuilder &builder) |
| |
| void | mlir::shard::maybeInsertSourceShardingAnnotation (Sharding sharding, OpOperand &operand, OpBuilder &builder) |
| |
| SmallVector< Value > | mlir::shard::getMixedAsValues (OpBuilder b, const Location &loc, llvm::ArrayRef< int64_t > statics, ValueRange dynamics, Type type=Type()) |
| | Converts a vector of OpFoldResults (ints) into vector of Values of the provided type. More...
|
| |
