aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/test/RuntimeTests.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/test/RuntimeTests.cpp')
-rw-r--r--src/armnn/test/RuntimeTests.cpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/test/RuntimeTests.cpp b/src/armnn/test/RuntimeTests.cpp
index 6b3fe0f211..427352aaac 100644
--- a/src/armnn/test/RuntimeTests.cpp
+++ b/src/armnn/test/RuntimeTests.cpp
@@ -26,6 +26,11 @@
#include "RuntimeTests.hpp"
#include <TestUtils.hpp>
+#ifdef ARMNN_LEAK_CHECKING_ENABLED
+#include <HeapProfiling.hpp>
+#include <LeakChecking.hpp>
+#endif
+
namespace armnn
{