diff --git a/opal/mca/common/ucx/common_ucx.c b/opal/mca/common/ucx/common_ucx.c index 0225231e32f..1cd250e8944 100644 --- a/opal/mca/common/ucx/common_ucx.c +++ b/opal/mca/common/ucx/common_ucx.c @@ -416,8 +416,10 @@ OPAL_DECLSPEC int opal_common_ucx_mca_pmix_fence(ucp_worker_h worker) return ret; } - while (!fenced) { - ucp_worker_progress(worker); + MCA_COMMON_UCX_PROGRESS_LOOP(worker) { + if(fenced) { + break; + } } return ret;