[v0.10] Add --list-gpus option

2021-08-23 22:39:31 +02:00
parent d00e3f183d
commit 4b4d1bc030
5 changed files with 73 additions and 16 deletions
--- a/src/cuda/cuda.hpp
+++ b/src/cuda/cuda.hpp
@@ -4,6 +4,7 @@
 #include "../common/gpu.hpp"

 struct gpu_info* get_gpu_info(int gpu_idx);
+int print_gpus_list();
 char* get_str_sm(struct gpu_info* gpu);
 char* get_str_cores_sm(struct gpu_info* gpu);
 char* get_str_cuda_cores(struct gpu_info* gpu);