[v0.21] Add verbose option. Fix CUDA driver initialization message when verbose output is used
This commit is contained in:
@@ -100,7 +100,7 @@ struct gpu_info* get_gpu_info_cuda(int gpu_idx) {
|
||||
}
|
||||
|
||||
if(gpu_idx == 0) {
|
||||
printf("Waiting for CUDA driver to start...");
|
||||
printf("%s", CUDA_DRIVER_START_WARNING);
|
||||
fflush(stdout);
|
||||
}
|
||||
|
||||
@@ -113,6 +113,7 @@ struct gpu_info* get_gpu_info_cuda(int gpu_idx) {
|
||||
|
||||
if(gpu_idx == 0) {
|
||||
printf("\r");
|
||||
fflush(stdout);
|
||||
}
|
||||
|
||||
if(num_gpus <= 0) {
|
||||
|
||||
@@ -2,6 +2,7 @@
|
||||
#define __CUDA_GPU__
|
||||
|
||||
#include "../common/gpu.hpp"
|
||||
#define CUDA_DRIVER_START_WARNING "Waiting for CUDA driver to start..."
|
||||
|
||||
struct gpu_info* get_gpu_info_cuda(int gpu_idx);
|
||||
bool print_gpu_cuda(struct gpu_info* gpu);
|
||||
|
||||
Reference in New Issue
Block a user