Skip to content

Commit ae1a54d

Browse files
author
Sergey Oblomov
committed
PML/UCX: fixed hand on MPI_Finalize
fixes issue openucx/ucx#2656 added flush for worker object to complete all pending operations Signed-off-by: Sergey Oblomov <sergeyo@mellanox.com> (cherry picked from commit 0a8261f)
1 parent 15e8092 commit ae1a54d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

ompi/mca/pml/ucx/pml_ucx.c

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -435,6 +435,10 @@ int mca_pml_ucx_del_procs(struct ompi_proc_t **procs, size_t nprocs)
435435

436436
mca_pml_ucx_waitall(dreqs, &num_reqs);
437437
free(dreqs);
438+
/* flush worker to allow all pending operations to complete.
439+
* ignore error (we can do nothing here), just try to
440+
* finalize gracefully */
441+
ucp_worker_flush(ompi_pml_ucx.ucp_worker);
438442

439443
opal_pmix.fence(NULL, 0);
440444

0 commit comments

Comments
 (0)