We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dfdc219 commit 0cb1a33Copy full SHA for 0cb1a33
src/accelerate/state.py
@@ -400,7 +400,7 @@ def wait_for_everyone(self):
400
DistributedType.DEEPSPEED,
401
DistributedType.FSDP,
402
):
403
- torch.distributed.barrier(device_ids=[self.process_index])
+ torch.distributed.barrier(device_ids=[self.local_process_index])
404
elif self.distributed_type == DistributedType.XLA:
405
xm.rendezvous("accelerate.utils.wait_for_everyone")
406
0 commit comments