UCT/CUDA/CUDA_IPC: Handle cuMemRelease error status. (#10827)

Author: rakhmets

src/uct/cuda/cuda_ipc/cuda_ipc_cache.c

@@ -261,18 +261,14 @@ uct_cuda_ipc_open_memhandle_vmm(uct_cuda_ipc_rkey_t *key, CUdevice cu_dev,
     }
 
     *mapped_addr = dptr;
-
-    status = UCT_CUDADRV_FUNC_LOG_ERR(cuMemRelease(handle));
-    if (status == UCS_OK) {
-        goto out;
-    }
+    goto release_handle;
 
 unmap_range:
-    cuMemUnmap(dptr, key->b_len);
+    UCT_CUDADRV_FUNC_LOG_WARN(cuMemUnmap(dptr, key->b_len));
 release_va_range:
-    cuMemAddressFree(dptr, key->b_len);
+    UCT_CUDADRV_FUNC_LOG_WARN(cuMemAddressFree(dptr, key->b_len));
 release_handle:
-    cuMemRelease(handle);
+    UCT_CUDADRV_FUNC_LOG_WARN(cuMemRelease(handle));
 out:
     return status;
 }

src/uct/cuda/cuda_ipc/cuda_ipc_md.c

@@ -184,17 +184,12 @@ uct_cuda_ipc_mem_add_reg(void *addr, uct_cuda_ipc_memh_t *memh,
             UCT_CUDADRV_FUNC_LOG_ERR(cuMemExportToShareableHandle(
                         &key->ph.handle.fabric_handle, handle,
                         CU_MEM_HANDLE_TYPE_FABRIC, 0));
+        UCT_CUDADRV_FUNC_LOG_WARN(cuMemRelease(handle));
         if (status != UCS_OK) {
-            cuMemRelease(handle);
             ucs_debug("unable to export handle for VMM ptr: %p", addr);
             goto non_ipc;
         }
 
-        status = UCT_CUDADRV_FUNC_LOG_ERR(cuMemRelease(handle));
-        if (status != UCS_OK) {
-            goto out_pop_ctx;
-        }
-
         key->ph.handle_type = UCT_CUDA_IPC_KEY_HANDLE_TYPE_VMM;
         ucs_trace("packed vmm fabric handle for %p", addr);
         goto common_path;