Merge branch 'master' of github.com:liuq901/nerv into liuq901-master

1 files changed, 5 insertions, 1 deletions

diff --git a/nerv/lib/matrix/cumatrix.c b/nerv/lib/matrix/cumatrix.c
index aec4d60..43448bf 100644
--- a/nerv/lib/matrix/cumatrix.c
+++ b/nerv/lib/matrix/cumatrix.c
@@ -9,15 +9,19 @@
 void nerv_cuda_context_print_profile(CuContext *context) {
     HashMap *profile = context->profile;
     size_t i;
+    float tmp, tot = 0;
     fprintf(stderr, "*** [nerv cumatrix profile] **\n");
     for (i = 0; i < profile->size; i++)
     {
         HashNode *ptr;
         for (ptr = profile->bucket[i]; ptr; ptr = ptr->next)
         {
-            fprintf(stderr, "%s:\t%.6f\n", ptr->key, *(float *)ptr->val);
+            tmp = *(float *)ptr->val;
+            fprintf(stderr, "%s:\t%.6f\n", ptr->key, tmp);
+            tot += tmp;
         }
     }
+    fprintf(stderr, "Total time:\t%.6f\n", tot);
 }
 
 void nerv_cuda_context_clear_profile(CuContext *context) {