show memory usage

source/tensor/XMem.cpp

@@ -1461,6 +1461,23 @@ void XMem::CreateBLASHandle()
 #endif
 #endif
 }
 }
 
 
+/* show profile of the memory pool */
+void XMem::ShowMemUsage(FILE * file)
+{
+    MTYPE used = 0;
+    MTYPE total = 0;
+
+    for(int i = 0; i < blockNum; i++){
+        if(blocks[i].mem != NULL){
+            used  += blocks[i].used;
+            total += blocks[i].size;
+        }
+    }
+
+    fprintf(file, "mem:%.1fMB used:%.1fMB usage:%.3f\n", 
+           (DTYPE)used/MILLION, (DTYPE)total/MILLION, (DTYPE)used/total);
+}
+
 #ifdef USE_CUDA
 #ifdef USE_CUDA
 
 
 /* get the handle of cublas */
 /* get the handle of cublas */

source/tensor/XMem.h

@@ -24,6 +24,7 @@
 #ifndef __XMEM_H__
 #ifndef __XMEM_H__
 #define __XMEM_H__
 #define __XMEM_H__
 
 
+#include <stdio.h>
 #include <stdlib.h>
 #include <stdlib.h>
 
 
 #ifdef CUDA_BLAS
 #ifdef CUDA_BLAS
@@ -402,6 +403,9 @@ public:
     /* create a new cublas handle */
     /* create a new cublas handle */
     void CreateBLASHandle();
     void CreateBLASHandle();
 
 
+    /* show profile of the memory pool */
+    void ShowMemUsage(FILE * file);
+
 #ifdef USE_CUDA
 #ifdef USE_CUDA
     /* get the handle of cublas */
     /* get the handle of cublas */
     cublasHandle_t * GetCublasHandle();
     cublasHandle_t * GetCublasHandle();