[v0.10] Simple refactoring

src/cuda/cuda.cpp

@@ -160,31 +160,27 @@ struct gpu_info* get_gpu_info(int gpu_idx) {
   return gpu;
 }
 
-char* get_str_sm(struct gpu_info* gpu) {
+char* get_str_generic(int32_t data) {
-  uint32_t max_size = 10;
+  // Largest int is 10, +1 for possible negative, +1 for EOL
+  uint32_t max_size = 12;
   char* dummy = (char *) ecalloc(max_size, sizeof(char));
-  snprintf(dummy, max_size, "%d", gpu->topo->streaming_mp);
+  snprintf(dummy, max_size, "%d", data);
   return dummy;
 }
 
+char* get_str_sm(struct gpu_info* gpu) {
+  return get_str_generic(gpu->topo->streaming_mp);
+}
+
 char* get_str_cores_sm(struct gpu_info* gpu) {
-  uint32_t max_size = 10;
+  return get_str_generic(gpu->topo->cores_per_mp);
-  char* dummy = (char *) ecalloc(max_size, sizeof(char));
-  snprintf(dummy, max_size, "%d", gpu->topo->cores_per_mp);
-  return dummy;
 }
 
 char* get_str_cuda_cores(struct gpu_info* gpu) {
-  uint32_t max_size = 10;
+  return get_str_generic(gpu->topo->cuda_cores);
-  char* dummy = (char *) ecalloc(max_size, sizeof(char));
-  snprintf(dummy, max_size, "%d", gpu->topo->cuda_cores);
-  return dummy;
 }
 
 char* get_str_tensor_cores(struct gpu_info* gpu) {
-  uint32_t max_size = 10;
+  return get_str_generic(gpu->topo->tensor_cores);
-  char* dummy = (char *) ecalloc(max_size, sizeof(char));
-  snprintf(dummy, max_size, "%d", gpu->topo->tensor_cores);
-  return dummy;
 }