Browse Source

Merge pull request #2209 from dhiltgen/harden_mgmt

Fix crash on cuda ml init failure
Daniel Hiltgen 1 year ago
parent
commit
23a7ea593b
1 changed files with 1 additions and 0 deletions
  1. 1 0
      gpu/gpu_info_cuda.c

+ 1 - 0
gpu/gpu_info_cuda.c

@@ -70,6 +70,7 @@ void cuda_init(char *cuda_lib_path, cuda_init_resp_t *resp) {
     resp->ch.handle = NULL;
     resp->ch.handle = NULL;
     snprintf(buf, buflen, "nvml vram init failure: %d", ret);
     snprintf(buf, buflen, "nvml vram init failure: %d", ret);
     resp->err = strdup(buf);
     resp->err = strdup(buf);
+    return;
   }
   }
 
 
   // Report driver version if we're in verbose mode, ignore errors
   // Report driver version if we're in verbose mode, ignore errors