cuda : print the returned error when CUDA initialization fails (#6185)

slaren · web-flow · commit 1c51f98adcba · 2024-03-20T21:03:26.000+01:00
diff --git a/ggml-cuda.cu b/ggml-cuda.cu
@@ -294,8 +294,9 @@ static ggml_cuda_device_info ggml_cuda_init() {
 
     ggml_cuda_device_info info = {};
 
-    if (cudaGetDeviceCount(&info.device_count) != cudaSuccess) {
-        fprintf(stderr, "%s: no " GGML_CUDA_NAME " devices found, " GGML_CUDA_NAME " will be disabled\n", __func__);
+    cudaError_t err = cudaGetDeviceCount(&info.device_count);
+    if (err != cudaSuccess) {
+        fprintf(stderr, "%s: failed to initialize " GGML_CUDA_NAME ": %s\n", __func__, cudaGetErrorString(err));
         return info;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -294,8 +294,9 @@ static ggml_cuda_device_info ggml_cuda_init() {`
`294`	`294`
`295`	`295`	`ggml_cuda_device_info info = {};`
`296`	`296`
`297`		`- if (cudaGetDeviceCount(&info.device_count) != cudaSuccess) {`
`298`		`- fprintf(stderr, "%s: no " GGML_CUDA_NAME " devices found, " GGML_CUDA_NAME " will be disabled\n", __func__);`
	`297`	`+ cudaError_t err = cudaGetDeviceCount(&info.device_count);`
	`298`	`+ if (err != cudaSuccess) {`
	`299`	`+ fprintf(stderr, "%s: failed to initialize " GGML_CUDA_NAME ": %s\n", __func__, cudaGetErrorString(err));`
`299`	`300`	`return info;`
`300`	`301`	`}`
`301`	`302`