| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | MeshShardingInterfaceImpl.cpp | 172 createDestinationPassingStyleInitOperands(LinalgOp op,MeshOp meshOp,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshAxis> reductionMeshAxes,IRMapping & spmdizationMap,ImplicitLocOpBuilder & builder) createDestinationPassingStyleInitOperands() argument
189 createAllReduceForResultWithoutPartialSharding(Value unshardedLinalgOpResult,ArrayRef<MeshAxis> opReductionMeshAxes,MeshShardingAttr resultSharding,ReductionKind reductionKind,IRMapping & spmdizationMap,ImplicitLocOpBuilder & builder) createAllReduceForResultWithoutPartialSharding() argument
209 createAllReduceForResultsWithoutPartialShardings(LinalgOp unshardedOp,ArrayRef<MeshAxis> opReductionMeshAxes,ArrayRef<MeshShardingAttr> resultShardings,IRMapping & spmdizationMap,ImplicitLocOpBuilder & builder) createAllReduceForResultsWithoutPartialShardings() argument
226 spmdizeLinalgOpWithShardedReduction(LinalgOp op,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshShardingAttr> operandShardings,ArrayRef<MeshShardingAttr> resultShardings,ArrayRef<utils::IteratorType> loopIteratorTypes,ArrayRef<SmallVector<MeshAxis>> meshAxisAssignmentForLoopIterators,IRMapping & spmdizationMap,SymbolTableCollection & symbolTable,ImplicitLocOpBuilder & builder) spmdizeLinalgOpWithShardedReduction() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ |
| H A D | Spmdization.cpp | 549 spmdizeOperation(Operation & op,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshShardingAttr> operandShardings,ArrayRef<MeshShardingAttr> resultShardings,IRMapping & spmdizationMap,SymbolTableCollection & symbolTableCollection,OpBuilder & builder) spmdizeOperation() argument
615 spmdizeOperation(ShardOp shardOp,IRMapping & spmdizationMap,SymbolTableCollection & symbolTableCollection,OpBuilder & builder) spmdizeOperation() argument
640 spmdizeOperation(Operation & op,IRMapping & spmdizationMap,SymbolTableCollection & symbolTableCollection,OpBuilder & builder) spmdizeOperation() argument
660 spmdizeBlock(Block & block,IRMapping & spmdizationMap,SymbolTableCollection & symbolTableCollection,OpBuilder & builder) spmdizeBlock() argument
687 spmdizeFuncOp(FunctionOpInterface op,IRMapping & spmdizationMap,SymbolTableCollection & symbolTableCollection) spmdizeFuncOp() argument
733 IRMapping spmdizationMap; runOnOperation() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Mesh/Interfaces/ |
| H A D | ShardingInterface.cpp | 594 spmdizeFullyReplicatedOperation(Operation & op,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshShardingAttr> operandShardings,ArrayRef<MeshShardingAttr> resultShardings,IRMapping & spmdizationMap,SymbolTableCollection & symbolTable,OpBuilder & builder) spmdizeFullyReplicatedOperation() argument
690 spmdizeTriviallyShardableOperation(Operation & op,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshShardingAttr> operandShardings,ArrayRef<MeshShardingAttr> resultShardings,IRMapping & spmdizationMap,SymbolTableCollection & symbolTable,OpBuilder & builder) spmdizeTriviallyShardableOperation() argument
|