From 5d9c4a5f5a5d6aab0e8b4aedf504c6a0e526b2f4 Mon Sep 17 00:00:00 2001
From: Daniel Hiltgen <daniel@ollama.com>
Date: Fri, 26 Jan 2024 09:18:33 -0800
Subject: [PATCH] Fix crash on cuda ml init failure

The new driver lookup code was triggering after init failure due to a missing return
---
 gpu/gpu_info_cuda.c | 1 +
 1 file changed, 1 insertion(+)

diff --git a/gpu/gpu_info_cuda.c b/gpu/gpu_info_cuda.c
index 9299b22c..d877ff0c 100644
--- a/gpu/gpu_info_cuda.c
+++ b/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;
     snprintf(buf, buflen, "nvml vram init failure: %d", ret);
     resp->err = strdup(buf);
+    return;
   }
 
   // Report driver version if we're in verbose mode, ignore errors