There was an error while loading. Please reload this page.
1 parent b390dd5 commit 1ddbb1bCopy full SHA for 1ddbb1b
1 file changed
torch_xla/csrc/xla_sharding_util.cpp
@@ -759,13 +759,13 @@ void ShardingUtil::ReshardParameters(
759
std::vector<int64_t> denormalized_tile_assignment;
760
auto sharding_spec = (*tensors)[0]->sharding_spec();
761
if (sharding_spec) {
762
- denormalized_tile_assignment = sharding_spec->sharding.GetDenormalizedTileAssignment();
+ denormalized_tile_assignment =
763
+ sharding_spec->sharding.GetDenormalizedTileAssignment();
764
}
765
for (const auto& sharding : xla_input_shardings) {
- if (denormalized_tile_assignment.size() > 0){
766
+ if (denormalized_tile_assignment.size() > 0) {
767
input_shardings.emplace_back(sharding, denormalized_tile_assignment);
- }
768
- else{
+ } else {
769
input_shardings.emplace_back(sharding);
770
771
0 commit comments