Explorar o código

cuda : print message when initialization fails (#5512)

* cuda : print message when initialization fails

* use CUDA_NAME both times
slaren hai 1 ano
pai
achega
9060a1e9df
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  1. 1 0
      ggml-cuda.cu

+ 1 - 0
ggml-cuda.cu

@@ -7943,6 +7943,7 @@ GGML_CALL void ggml_init_cublas() {
         if (cudaGetDeviceCount(&g_device_count) != cudaSuccess) {
             initialized = true;
             g_cublas_loaded = false;
+            fprintf(stderr, "%s: no " GGML_CUDA_NAME " devices found, " GGML_CUDA_NAME " will be disabled\n", __func__);
             return;
         }