Skip to content

Commit e04575e

Browse files
authored
Merge pull request #7274 from janjust/master-oshmem-perf-multi-worker
oshmem/ucx: multi-threaded spml ucx performance improvements
2 parents 611a2bd + 3d6bf9f commit e04575e

File tree

5 files changed

+213
-102
lines changed

5 files changed

+213
-102
lines changed

oshmem/mca/atomic/ucx/atomic_ucx_cswap.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,6 @@ int mca_atomic_ucx_cswap(shmem_ctx_t ctx,
5050
mca_spml_ucx_remote_op_posted(ucx_ctx, pe);
5151
}
5252

53-
return opal_common_ucx_wait_request(status_ptr, ucx_ctx->ucp_worker,
53+
return opal_common_ucx_wait_request(status_ptr, ucx_ctx->ucp_worker[0],
5454
"ucp_atomic_fetch_nb");
5555
}

oshmem/mca/atomic/ucx/atomic_ucx_module.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ int mca_atomic_ucx_fop(shmem_ctx_t ctx,
8080
op, value, prev, size,
8181
rva, ucx_mkey->rkey,
8282
opal_common_ucx_empty_complete_cb);
83-
return opal_common_ucx_wait_request(status_ptr, ucx_ctx->ucp_worker,
83+
return opal_common_ucx_wait_request(status_ptr, ucx_ctx->ucp_worker[0],
8484
"ucp_atomic_fetch_nb");
8585
}
8686

0 commit comments

Comments
 (0)