Method barrier

checkpoint_engine/distributed/base.py:149–150

(self, group: DistributedProcessGroup | None = None, **kwargs)

        torch_dist.broadcast(tensor, src, group, **kwargs)

    def barrier(self, group: DistributedProcessGroup | None = None, **kwargs):
        torch_dist.barrier(group, **kwargs)

    def new_group(self, ranks: list[int], **kwargs) -> DistributedProcessGroup | None:
        return torch_dist.new_group(ranks, **kwargs)