From d4efb6a2a56401f98ca2c0756005fd2c7d7bfce5 Mon Sep 17 00:00:00 2001 From: Logan Adams <114770087+loadams@users.noreply.github.com> Date: Mon, 24 Feb 2025 18:45:11 -0800 Subject: [PATCH] Revert "Handle special case of libuv for Windows (#7064)" (#7076) This reverts commit 8577bd244903f454e319953a6ae1c3838558ac69. Fixes: #7072 Signed-off-by: Max Kovalenko --- 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