From 951b0c3c37d2a974ad8972f79a817182cf5224b6 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Mon, 24 Feb 2025 11:31:26 -0800 Subject: [PATCH] Revert "Handle special case of libuv for Windows (#7064)" This reverts commit 8577bd244903f454e319953a6ae1c3838558ac69. --- deepspeed/comm/torch.py | 21 +++++---------------- 1 file changed, 5 insertions(+), 16 deletions(-) diff --git a/deepspeed/comm/torch.py b/deepspeed/comm/torch.py index 1146832d7655..efa0640fb87b 100755 --- a/deepspeed/comm/torch.py +++ b/deepspeed/comm/torch.py @@ -145,22 +145,11 @@ def has_reduce_scatter_tensor(self): def init_process_group(self, backend, timeout, init_method, rank, world_size): if not torch.distributed.is_initialized(): - if not required_torch_version(min_version=2.4): - # Windows torch builds do not come with lib_uv by default. - # More information here: https://pytorch.org/tutorials/intermediate/TCPStore_libuv_backend.html - use_libuv = False if os.name == "nt" else True - torch.distributed.init_process_group(backend, - timeout=timeout, - init_method=init_method, - rank=rank, - world_size=world_size, - use_libuv=use_libuv) - else: - torch.distributed.init_process_group(backend, - timeout=timeout, - init_method=init_method, - rank=rank, - world_size=world_size) + torch.distributed.init_process_group(backend, + timeout=timeout, + init_method=init_method, + rank=rank, + world_size=world_size) self.using_mpi = torch.distributed.get_backend() == 'mpi' @disable_compiler_collective