Skip to content

Commit b8e1af6

Browse files
committed
osc_ucx: add worker flush before osc module free
Make sure all pending communications are done on all ranks before closing the window. This way it will be safe to close the endpoints when closing the component. Signed-off-by: Yossi Itigin <yosefe@mellanox.com>
1 parent bcc4851 commit b8e1af6

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ompi/mca/osc/ucx/osc_ucx_component.c

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -825,6 +825,8 @@ int ompi_osc_ucx_free(struct ompi_win_t *win) {
825825
ucp_worker_progress(mca_osc_ucx_component.ucp_worker);
826826
}
827827

828+
opal_common_ucx_worker_flush(mca_osc_ucx_component.ucp_worker);
829+
828830
ret = module->comm->c_coll->coll_barrier(module->comm,
829831
module->comm->c_coll->coll_barrier_module);
830832

0 commit comments

Comments
 (0)