UCP/TEST: fix flush

Author: shasson5

src/ucp/rma/flush.c

@@ -530,7 +530,7 @@ UCS_PROFILE_FUNC(ucs_status_ptr_t, ucp_ep_flush_nbx, (ep, param),
     return request;
 }
 
-static ucs_status_t ucp_worker_flush_check(ucp_worker_h worker)
+ucs_status_t ucp_worker_flush_check(ucp_worker_h worker)
 {
     ucp_rsc_index_t iface_id;
     ucp_worker_iface_t *wiface;

src/ucp/rma/rma.h

@@ -100,6 +100,7 @@ extern ucp_amo_proto_t ucp_amo_sw_proto;
 extern const ucp_rma_proto_t *ucp_rma_proto_list[];
 extern const ucp_amo_proto_t *ucp_amo_proto_list[];
 
+ucs_status_t ucp_worker_flush_check(ucp_worker_h worker);
 
 ucs_status_t ucp_rma_request_advance(ucp_request_t *req, ssize_t frag_length,
                                      ucs_status_t status,

test/gtest/ucp/test_ucp_memheap.cc

@@ -10,6 +10,10 @@
 #include <common/mem_buffer.h>
 #include <common/test_helpers.h>
 #include <ucs/sys/sys.h>
+extern "C" {
+#include <ucp/rma/rma.h>
+}
+
 #include <ucs/sys/ptr_arith.h>
 
 
@@ -95,6 +99,9 @@ void test_ucp_memheap::test_xfer(send_func_t send_func, size_t size,
         flush_ep(sender());
     } else {
         flush_worker(sender());
+        while(ucp_worker_flush_check(sender().worker()) != UCS_OK) {
+            progress();
+        }
     }
 
     /* Validate data */