From 4d948eb80a092e41002b4729ad3d207e3d3254ec Mon Sep 17 00:00:00 2001
From: Dr-Noob <peibolms@gmail.com>
Date: Sat, 21 May 2022 23:19:03 +0200
Subject: [PATCH] [v0.24] Remove CUDA driver initialization message before
 printing any other message

---
 src/cuda/cuda.cpp | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

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