Remove redundant cod and fix compilation error

Signed-off-by: Michal Shalev <[email protected]>

Author: michal-shalev

src/plugins/ucx/ucx_backend.cpp

@@ -1344,7 +1344,7 @@ void* nixlUcxEngine::exportXferReqtoGPU(nixlBackendReqH* handle) const {
         return nullptr;
     }
 
-    return ep->exported_batch;
+    return ep->exported_batches.at(ucx_req);
 }
 
 nixl_status_t nixlUcxEngine::releaseXferReqtoGPU(nixlBackendReqH* handle) const {

src/utils/ucx/ucx_utils.cpp

@@ -399,10 +399,9 @@ nixl_status_t nixlUcxEp::exportBatch(nixlUcxReq &req)
         return NIXL_ERR_INVALID_PARAM;
     }
 
-    // Clear any existing batch
-    exported_batch = nullptr;
+    void *exported_batch = nullptr;
 
-    ucs_status_t status = ucp_ep_rma_batch_export((void*)req, (ucp_batch_h*)&exported_batch);
+    ucs_status_t status = ucp_ep_rma_batch_export((void *)req, (ucp_batch_h *)&exported_batch);
     if (status != UCS_OK) {
         NIXL_ERROR << "Failed to export batch request: " << ucs_status_string(status);
         exported_batch = nullptr;
@@ -420,7 +419,7 @@ nixl_status_t nixlUcxEp::releaseBatch(nixlUcxReq &req)
         return NIXL_ERR_INVALID_PARAM;
     }
 
-    ucs_status_t status = ucp_ep_rma_batch_release((void *)req, exported_batches[req]);
+    ucs_status_t status = ucp_ep_rma_batch_release((void *)req, (ucp_batch_h)exported_batches[req]);
     if (status != UCS_OK) {
         NIXL_ERROR << "Failed to release UCX batch: " << ucs_status_string(status);
     }

src/utils/ucx/ucx_utils.h

@@ -86,8 +86,7 @@ class nixlUcxEp {
     nixl_status_t disconnect_nb();
 public:
     void err_cb(ucp_ep_h ucp_ep, ucs_status_t status);
-    void* exported_batch{nullptr};
-    std::unordered_map<nixlUcxReq, ucp_batch_h> exported_batches;
+    std::unordered_map<nixlUcxReq, void *> exported_batches;
 
     nixl_status_t checkTxState() const {
         switch (state) {