Skip cleanup when no device was allocated

Mellanox · Nov 27, 2023 · 4451375 · 4451375
1 parent 5a8a6d6
commit 4451375
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/ucx_plugin.c b/src/ucx_plugin.c
@@ -420,9 +420,11 @@ static ncclResult_t nccl_ucx_free_worker(nccl_ucx_worker_t *ucx_worker) {
       free(ucx_worker);
     }
 
-    ucp_cleanup(ucp_ctx[dev]);
     workers[dev] = NULL;
-    ucp_ctx[dev] = NULL;
+    if (ucp_ctx[dev]) {
+      ucp_cleanup(ucp_ctx[dev]);
+      ucp_ctx[dev] = NULL;
+    }
   }
 
   return ncclSuccess;