Explorar el Código

cuda : print message when initialization fails (#5512)

* cuda : print message when initialization fails

* use CUDA_NAME both times
slaren hace 1 año
padre
commit
9060a1e9df
Se han modificado 1 ficheros con 1 adiciones y 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;
         }