diff --git a/config/ompi_check_ucx.m4 b/config/ompi_check_ucx.m4 index c1e540dd83..e5a5ccf047 100644 --- a/config/ompi_check_ucx.m4 +++ b/config/ompi_check_ucx.m4 @@ -108,7 +108,8 @@ AC_DEFUN([OMPI_CHECK_UCX],[ [AC_DEFINE([HAVE_UCP_TAG_SEND_NBR],[1], [have ucp_tag_send_nbr()])], [], [#include ]) - AC_CHECK_DECLS([ucp_ep_flush_nb, ucp_worker_flush_nb, ucp_request_check_status, ucp_put_nb, ucp_get_nb], + AC_CHECK_DECLS([ucp_ep_flush_nb, ucp_worker_flush_nb, + ucp_request_check_status, ucp_put_nb, ucp_get_nb], [], [], [#include ]) CPPFLAGS=$old_CPPFLAGS diff --git a/oshmem/mca/spml/ucx/spml_ucx.c b/oshmem/mca/spml/ucx/spml_ucx.c index 4f7fd80ec0..555d36eade 100644 --- a/oshmem/mca/spml/ucx/spml_ucx.c +++ b/oshmem/mca/spml/ucx/spml_ucx.c @@ -604,7 +604,7 @@ int mca_spml_ucx_put(void* dst_addr, size_t size, void* src_addr, int dst) /* TODO: replace wait_request by opal_common_ucx_wait_request_opal_status */ status = opal_common_ucx_wait_request(request, mca_spml_ucx.ucp_worker); #else - request = ucp_put(mca_spml_ucx.ucp_peers[dst].ucp_conn, src_addr, size, + status = ucp_put(mca_spml_ucx.ucp_peers[dst].ucp_conn, src_addr, size, (uint64_t)rva, ucx_mkey->rkey); #endif return ucx_status_to_oshmem(status);