Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors.

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-01 19:18:43 +02:00
parent c06c9fcf23
commit af13e7c954
14 changed files with 255 additions and 128 deletions
@@ -12,6 +12,10 @@
 #include <cublas_v2.h>
 #include <cudnn.h>

+#include <unistd.h>
+#include <ios>
+
+
 #define dnnType float


@@ -102,4 +106,6 @@ void matrixTranspose(cublasHandle_t handle, dnnType* srcData, dnnType* dstData,

 void matrixMulAdd(  cublasHandle_t handle, dnnType* srcData, dnnType* dstData, 
                    dnnType* add_vector, int dim, dnnType mul);
+
+void getMemUsage(double& vm_usage_kb, double& resident_set_kb);
 #endif //UTILS_H