blob: 54177b81757fd3e1730e545a31e94f1fd9d5052a (
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
39
40
41
42
43
44
45
46
47
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
#ifdef ARMNN_HEAP_PROFILING_ENABLED
#include <string>
#include <cstddef>
// This is conditional so we can change the environment variable
// at build time.
#ifndef ARMNN_HEAP_PROFILE_DUMP_DIR
#define ARMNN_HEAP_PROFILE_DUMP_DIR "ARMNN_HEAP_PROFILE_DUMP_DIR"
#endif // ARMNN_HEAP_PROFILE_DUMP_DIR
namespace armnnUtils
{
class ScopedHeapProfiler final
{
public:
ScopedHeapProfiler(const std::string & tag);
~ScopedHeapProfiler();
private:
// Location comes from the ARMNN_HEAP_PROFILE_DUMP_DIR.
// If it is not available then it dumps to /tmp.
std::string m_Location;
std::string m_Tag;
// No default construction and copying.
ScopedHeapProfiler() = delete;
ScopedHeapProfiler(const ScopedHeapProfiler &) = delete;
ScopedHeapProfiler & operator=(const ScopedHeapProfiler &) = delete;
};
} // namespace armnnUtils
#define ARMNN_SCOPED_HEAP_PROFILING(TAG) \
armnnUtils::ScopedHeapProfiler __scoped_armnn_heap_profiler__(TAG)
#else // ARMNN_HEAP_PROFILING_ENABLED
#define ARMNN_SCOPED_HEAP_PROFILING(TAG)
#endif // ARMNN_HEAP_PROFILING_ENABLED
|
