Просмотр исходного кода

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

slaren 1 год назад
Родитель
Сommit
1c51f98adc
1 измененных файлов с 3 добавлено и 2 удалено
  1. 3 2
      ggml-cuda.cu

+ 3 - 2
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;
     }