[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/common/gpu.hpp
+++ b/src/common/gpu.hpp
@@ -40,6 +40,7 @@ struct topology {
  int32_t streaming_mp;
  int32_t cores_per_mp;
  int32_t cuda_cores;
+  int32_t tensor_cores;
 };

 struct memory {