Fix indent for line 317 (shard.grad = grad_shard)

pytorch · Mar 19, 2024 · 173ef47 · 173ef47
1 parent ae348b2
commit 173ef47
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch_xla/distributed/zero_redundancy_optimizer.py b/torch_xla/distributed/zero_redundancy_optimizer.py
@@ -314,7 +314,8 @@ def step(self, closure=None, **kwargs):
 
             if grad_shard.dtype != self.optimizer_dtype:
               grad_shard = grad_shard.to(dtype=self.optimizer_dtype)
-          shard.grad = grad_shard
+            shard.grad = grad_shard
+
           index += 1
 
     if self.grad_clipping: