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
diff --git a/src/armnn/test/RuntimeTests.cpp b/src/armnn/test/RuntimeTests.cpp
index 6b3fe0f..427352a 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
{