Merge pull request #8 from dmitrygx/topic/fix_request_leak

TEST/APPS: Fix UCP request leak in case of unsuccessful ucp_stream_send_nb
yosefe · Apr 28, 2020 · 2ba6834 · 2ba6834
2 parents 5d6c73d + c08f82c
commit 2ba6834
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/apps/iodemo/ucx_wrapper.cc b/test/apps/iodemo/ucx_wrapper.cc
@@ -594,6 +594,8 @@ bool UcxConnection::connect_common(ucp_ep_params_t& ep_params)
                                     stream_send_callback, 0);
     if (!_context.wait_completion(sreq, 10)) {
         UCX_CONN_LOG << "failed to send remote connection id";
+        ucp_request_cancel(_context.worker(), rreq);
+        ucp_request_release(rreq);
         ep_close(UCP_EP_CLOSE_MODE_FORCE);
         return false;
     }