Merge pull request #193 from Jittor/ygy5

fix memory profiler bug
This commit is contained in:
yang guo ye 2021-04-06 18:42:57 +08:00 committed by GitHub
commit 8243ebc5b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

@ -91,6 +91,9 @@ void MemoryProfiler::check() {
Var* var = (Var*)node;
if (var->mem_ptr != nullptr) {
vector<Stack> stacks = get_node_trace(var);
if (stacks.size() == 0) {
stacks.push_back(Stack());
}
std::stringstream stream;
stream << var;
live_vars.push_back(std::make_pair(std::make_pair(stream.str(), stacks), var->size));