Franscobec/gpufetch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[v0.24] Remove CUDA driver initialization message before printing any other message

Browse Source
This commit is contained in:
Dr-Noob
2022-05-21 23:19:03 +02:00
parent cf96628385
commit 4d948eb80a
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/cuda/cuda.cpp
View File

@@ -118,16 +118,18 @@ struct gpu_info* get_gpu_info_cuda(struct pci_dev *devices, int gpu_idx) {
int num_gpus = -1; int num_gpus = -1;
cudaError_t err = cudaSuccess; cudaError_t err = cudaSuccess;
if ((err = cudaGetDeviceCount(&num_gpus)) != cudaSuccess) { err = cudaGetDeviceCount(&num_gpus);
printErr("%s: %s", cudaGetErrorName(err), cudaGetErrorString(err));
return NULL;
}
if(gpu_idx == 0) { if(gpu_idx == 0) {
printf("\r"); printf("\r%*c\r", (int) strlen(CUDA_DRIVER_START_WARNING), ' ');
fflush(stdout); fflush(stdout);
} }
if(err != cudaSuccess) {
printErr("%s: %s", cudaGetErrorName(err), cudaGetErrorString(err));
return NULL;
}
if(num_gpus <= 0) { if(num_gpus <= 0) {
printErr("No CUDA capable devices found!"); printErr("No CUDA capable devices found!");
return NULL; return NULL;