blob: 0b4b28a1b9d28bda7741755733d11c96a79e596e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#ifdef ARMNN_HEAP_PROFILING_ENABLED
#include "HeapProfiling.hpp"
#include "gperftools/heap-profiler.h"
#include <sstream>
#include <cstdlib>
namespace armnnUtils
{
ScopedHeapProfiler::ScopedHeapProfiler(const std::string & tag)
: m_Location("/tmp")
, m_Tag(tag)
{
char * locationFromEnv = ::getenv(ARMNN_HEAP_PROFILE_DUMP_DIR);
if (locationFromEnv)
{
m_Location = locationFromEnv;
}
std::stringstream ss;
ss << m_Location << "/" << m_Tag << ".hprof";
HeapProfilerStart(ss.str().c_str());
HeapProfilerDump(m_Tag.c_str());
}
ScopedHeapProfiler::~ScopedHeapProfiler()
{
HeapProfilerDump(m_Tag.c_str());
}
} // namespace armnnUtils
#endif // ARMNN_HEAP_PROFILING_ENABLED
|
