Method barrier

lightx2v/server/services/distributed_utils.py:85–90

(self)

            logger.info(f"Rank {self.rank} distributed environment cleaned up")

    def barrier(self):
        if self.is_initialized and dist.is_initialized():
            if torch.cuda.is_available() and dist.get_backend() == "nccl":
                dist.barrier(device_ids=[torch.cuda.current_device()])
            else:
                dist.barrier()

    def is_rank_zero(self) -> bool:
        return self.rank == 0