[v0.04] Add option to select GPU

2021-08-16 13:38:54 +02:00
parent eecc040e98
commit c7a08e7fd0
6 changed files with 98 additions and 17 deletions
--- a/src/cuda/cuda.hpp
+++ b/src/cuda/cuda.hpp
@@ -3,7 +3,7 @@

 #include "../common/gpu.hpp"

-struct gpu_info* get_gpu_info();
+struct gpu_info* get_gpu_info(int gpu_idx);
 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);