From 35fedc5b0b86d7f8afc84614b1489771e2e39283 Mon Sep 17 00:00:00 2001
From: Mike Kelly <mike.kelly@arm.com>
Date: Wed, 5 Apr 2023 13:43:24 +0100
Subject: IVGCVSW-7672 Arm NN build failures in Main

 * Added required includes if ARMNN_LEAK_CHECKING_ENABLED.

Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I18bde66e8a1e19690747d7e05d10dbe0cedb1c1b
---
 src/armnn/test/RuntimeTests.cpp | 5 +++++
 1 file changed, 5 insertions(+)

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
 {
 
-- 
cgit v1.2.1