Missing :

microsoft · Apr 5, 2024 · b7d5e94 · b7d5e94
1 parent 6d81351
commit b7d5e94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/comm/torch.py b/deepspeed/comm/torch.py
@@ -29,7 +29,7 @@ def has_all_reduce_coalesced():
 
 
 def get_coalescing_manager(group, device, reqs, async_op):
-    if required_torch_version(min_version=2.0, max_version=2.0)
+    if required_torch_version(min_version=2.0, max_version=2.0):
         return torch.distributed.distributed_c10d._coalescing_manager(group, device=device, reqs=reqs)
     elif required_torch_version(min_version=2.1):
         return torch.distributed.distributed_c10d._coalescing_manager(group, device=device, async_ops=async_op)