Added GPU memory report (#714)

* Added GPU memory report
* Added fixes for CPU build and clang-format
* Fixing memory reporting for CPU builds

Author: iomaganaris

coreneuron/utils/memory_utils.cpp

@@ -25,6 +25,7 @@
 #include <fstream>
 #include <unistd.h>
 #include "coreneuron/utils/memory_utils.h"
+#include "coreneuron/utils/profile/cuda_profile.h"
 #include "coreneuron/mpi/nrnmpi.h"
 #include "coreneuron/mpi/core/nrnmpi.hpp"
 #include "coreneuron/apps/corenrn_parameters.hpp"
@@ -105,6 +106,11 @@ void report_mem_usage(const char* message, bool all_ranks) {
                mem_max,
                mem_min,
                mem_avg);
+#ifdef CORENEURON_ENABLE_GPU
+        if (corenrn_param.gpu) {
+            print_gpu_memory_usage();
+        }
+#endif
     }
     fflush(stdout);
 }

coreneuron/utils/profile/cuda_profile.cu

@@ -8,6 +8,7 @@
 
 #include "cuda_profiler_api.h"
 #include <stdio.h>
+#include <ostream>
 
 void print_gpu_memory_usage() {
     size_t free_byte;
@@ -23,7 +24,7 @@ void print_gpu_memory_usage() {
     double free_db = (double) free_byte;
     double total_db = (double) total_byte;
     double used_db = total_db - free_db;
-    printf("\n  => GPU MEMORY USAGE (MB) : Used = %f, Free = %f MB, Total = %f",
+    printf(" GPU Memory (MBs) : Used = %f, Free = %f, Total = %f\n",
            used_db / 1024.0 / 1024.0,
            free_db / 1024.0 / 1024.0,
            total_db / 1024.0 / 1024.0);