diff --git a/deepspeed/comm/comm.py b/deepspeed/comm/comm.py index 765cc91287ea..568211645f40 100644 --- 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):