Changeset c15085d for src/Common/Stats/Heap.cc

Timestamp:

Jul 3, 2020, 4:50:21 PM (4 years ago)

Author:

Fangren Yu <f37yu@…>

Branches:

ADT, arm-eh, ast-experimental, enum, forall-pointer-decay, jacob/cs343-translation, master, new-ast, new-ast-unique-expr, pthread-emulation, qualifiedEnum

Children:

aebf5b0

Parents:

276f105

Message:

tracing memory allocation of resolver passes

File:

• src/Common/Stats/Heap.cc (modified) (6 diffs)

src/Common/Stats/Heap.cc

@@ -53 +53 @@
                 const size_t passes_size = sizeof(passes) / sizeof(passes[0]);
                 size_t       passes_cnt = 1;
+
+                StatBlock    stacktrace_stats[100];
+                size_t       stacktrace_stats_count = 0;
+                bool         stacktrace_stats_enabled = true;
+
+                size_t       trace[1000];
+                const size_t stacktrace_max_depth = sizeof(trace) / sizeof(size_t);
+                size_t       stacktrace_depth;
+
+                size_t new_stacktrace_id(const char * const name) {
+                        stacktrace_stats[stacktrace_stats_count].name = name;
+                        return stacktrace_stats_count++;
+                }
+
+                void stacktrace_push(size_t id) {
+                        ++stacktrace_depth;
+                        assertf(stacktrace_depth < stacktrace_max_depth, "Stack trace too deep: increase size of array in Heap.cc");
+                        trace[stacktrace_depth] = id;
+                }
+
+                void stacktrace_pop() {
+                        assertf(stacktrace_depth > 0, "Invalid stack tracing operation: trace is empty");
+                        --stacktrace_depth;
+                }
 
                 void newPass( const char * const name ) {
@@ -116 +140 @@
                         for(size_t i = 0; i < passes_cnt; i++) {
                                 print(passes[i], nc, total_mallocs, total_frees, overall_peak);
+                        }
+
+                        print('-', nct);
+                        std::cerr << std::setw(nc) << "Trace";
+                        std::cerr << " |       Malloc Count |         Free Count |        Peak Allocs |" << std::endl;
+
+                        print('-', nct);
+                        for (size_t i = 0; i < stacktrace_stats_count; i++) {
+                                print(stacktrace_stats[i], nc, total_mallocs, total_frees, overall_peak);
                         }
                         print('-', nct);
@@ -188 +221 @@
                                                 = std::max(passes[passes_cnt - 1].peak_allocs, passes[passes_cnt - 1].n_allocs);
                                 }
+
+                                if ( stacktrace_stats_enabled && stacktrace_depth > 0) {
+                                        stacktrace_stats[trace[stacktrace_depth]].mallocs++;
+                                }
                                 return __malloc( size );
                         }
@@ -196 +233 @@
                                         passes[passes_cnt - 1].frees++;
                                         passes[passes_cnt - 1].n_allocs--;
+                                }
+                                if ( stacktrace_stats_enabled && stacktrace_depth > 0) {
+                                        stacktrace_stats[trace[stacktrace_depth]].frees++;
                                 }
                                 return __free( ptr );
@@ -208 +248 @@
                                                 = std::max(passes[passes_cnt - 1].peak_allocs, passes[passes_cnt - 1].n_allocs);
                                 }
+                                if ( stacktrace_stats_enabled && stacktrace_depth > 0) {
+                                        stacktrace_stats[trace[stacktrace_depth]].mallocs++;
+                                }
                                 return __calloc( nelem, size );
                         }
@@ -218 +261 @@
                                         passes[passes_cnt - 1].frees++;
                                 } // if
+                                if ( stacktrace_stats_enabled && stacktrace_depth > 0) {
+                                        stacktrace_stats[trace[stacktrace_depth]].mallocs++;
+                                        stacktrace_stats[trace[stacktrace_depth]].frees++;
+                                }
                                 return s;
                         }