accelerate
Specify device_ids in torch.distributed.barrier for PartialState
#3744
Merged
Merged