vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cuda : print the returned error when CUDA initialization fails

Browse source
This commit is contained in:
slaren 2024-03-20 20:00:24 +01:00
parent f9c7ba3447
commit 2f7f7c02d8
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ggml-cuda.cu
View file

@ -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;
}