Fixed deepspeed.comm.monitored_barrier call (microsoft#4496)

* Fixed deepspeed.comm.monitored_barrier call * Formatting --------- Co-authored-by: Logan Adams <[email protected]>
baodii · Nov 7, 2023 · 586c65f · 586c65f
1 parent 8aa5dab
commit 586c65f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/comm/comm.py b/deepspeed/comm/comm.py
@@ -416,7 +416,7 @@ def monitored_barrier(group=None,
                       log_name='monitored_barrier',
                       debug=get_caller_func()):
     global cdb
-    return cdb.barrier(group=group, timeout=timeout, wait_all_ranks=wait_all_ranks)
+    return cdb.monitored_barrier(group=group, timeout=timeout, wait_all_ranks=wait_all_ranks)
 
 
 def log_summary(show_straggler=False):