[v0.10] Add support for displaying the number of tensor cores

2021-11-23 18:09:13 +01:00
parent f212fb88d4
commit 821b6e760e
4 changed files with 22 additions and 0 deletions
--- a/src/cuda/cuda.hpp
+++ b/src/cuda/cuda.hpp
@@ -8,5 +8,6 @@ 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);
+char* get_str_tensor_cores(struct gpu_info* gpu);

 #endif