baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix crash on cuda ml init failure

Browse source 
The new driver lookup code was triggering after init failure due to a missing return
This commit is contained in:
Daniel Hiltgen 2024-01-26 09:18:33 -08:00
parent 197e420a97
commit 5d9c4a5f5a
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
gpu/gpu_info_cuda.c
View file

@ -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