@@ -477,10 +477,10 @@ reshardOn1DGrid(ImplicitLocOpBuilder &builder, GridOp grid,
477
477
return targetShard;
478
478
}
479
479
480
- TypedValue<ShapedType> reshard (ImplicitLocOpBuilder &builder, GridOp grid,
481
- Sharding sourceSharding , Sharding targetSharding ,
482
- TypedValue<ShapedType> sourceUnshardedValue,
483
- TypedValue<ShapedType> sourceShard) {
480
+ static TypedValue<ShapedType>
481
+ reshard (ImplicitLocOpBuilder &builder, GridOp grid , Sharding sourceSharding ,
482
+ Sharding targetSharding, TypedValue<ShapedType> sourceUnshardedValue,
483
+ TypedValue<ShapedType> sourceShard) {
484
484
// If source and destination sharding are the same, no need to do anything.
485
485
if (sourceSharding == targetSharding || (isFullReplication (sourceSharding) &&
486
486
isFullReplication (targetSharding))) {
@@ -535,7 +535,7 @@ using UnshardedToShardedValueMap = DenseMap<Value, Value>;
535
535
// Get the types of block arguments for an partitioned block.
536
536
// Reads the sharding annotations of the arguments to deduce the sharded types.
537
537
// Types that are not ranked tensors are left unchanged.
538
- SmallVector<Type>
538
+ static SmallVector<Type>
539
539
shardedBlockArgumentTypes (Block &block,
540
540
SymbolTableCollection &symbolTableCollection) {
541
541
SmallVector<Type> res;
0 commit comments