Преглед изворни кода

ggml: bypass code incompatible with CUDA < 11.1 (whisper/2020)

`cudaHostRegisterReadOnly` parameter was only introduced in CUDA 11.1

See this issue for more details:
https://github.com/ggerganov/examples/whisper/whisper.cpp/issues/2007
Slava Primenko пре 1 година
родитељ
комит
f77261a7c5
1 измењених фајлова са 4 додато и 0 уклоњено
  1. 4 0
      ggml-cuda.cu

+ 4 - 0
ggml-cuda.cu

@@ -2617,6 +2617,7 @@ GGML_CALL bool ggml_backend_cuda_register_host_buffer(void * buffer, size_t size
         return false;
         return false;
     }
     }
 
 
+#if CUDART_VERSION >= 11100
     cudaError_t err = cudaHostRegister(buffer, size, cudaHostRegisterPortable | cudaHostRegisterReadOnly);
     cudaError_t err = cudaHostRegister(buffer, size, cudaHostRegisterPortable | cudaHostRegisterReadOnly);
     if (err != cudaSuccess) {
     if (err != cudaSuccess) {
         // clear the error
         // clear the error
@@ -2627,6 +2628,9 @@ GGML_CALL bool ggml_backend_cuda_register_host_buffer(void * buffer, size_t size
         return false;
         return false;
     }
     }
     return true;
     return true;
+#else
+    return false;
+#endif
 }
 }
 
 
 GGML_CALL void ggml_backend_cuda_unregister_host_buffer(void * buffer) {
 GGML_CALL void ggml_backend_cuda_unregister_host_buffer(void * buffer) {