Skip to content

Commit 6bbfca7

Browse files
authored
Merge pull request #10958 from jjhursey/ucx-add-procs
Reduce the overhead of UCX add_procs with intercommunicators
2 parents ade062a + 4eedf85 commit 6bbfca7

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

ompi/mca/pml/ucx/pml_ucx.c

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -410,6 +410,11 @@ static ucp_ep_h mca_pml_ucx_add_proc_common(ompi_proc_t *proc)
410410
ucp_ep_h ep;
411411
int ret;
412412

413+
/* Do not add a new endpoint if we already created one */
414+
if (NULL != proc->proc_endpoints[OMPI_PROC_ENDPOINT_TAG_PML]) {
415+
return proc->proc_endpoints[OMPI_PROC_ENDPOINT_TAG_PML];
416+
}
417+
413418
ret = mca_pml_ucx_recv_worker_address(proc, &address, &addrlen);
414419
if (ret < 0) {
415420
return NULL;

0 commit comments

Comments
 (0)