File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -759,13 +759,13 @@ void ShardingUtil::ReshardParameters(
759
759
std::vector<int64_t > denormalized_tile_assignment;
760
760
auto sharding_spec = (*tensors)[0 ]->sharding_spec ();
761
761
if (sharding_spec) {
762
- denormalized_tile_assignment = sharding_spec->sharding .GetDenormalizedTileAssignment ();
762
+ denormalized_tile_assignment =
763
+ sharding_spec->sharding .GetDenormalizedTileAssignment ();
763
764
}
764
765
for (const auto & sharding : xla_input_shardings) {
765
- if (denormalized_tile_assignment.size () > 0 ){
766
+ if (denormalized_tile_assignment.size () > 0 ) {
766
767
input_shardings.emplace_back (sharding, denormalized_tile_assignment);
767
- }
768
- else {
768
+ } else {
769
769
input_shardings.emplace_back (sharding);
770
770
}
771
771
}
You can’t perform that action at this time.
0 commit comments