From: Finn Williams <Finn.Williams@arm.com>
Date: Wed, 18 Mar 2020 14:20:37 +0000 (+0000)
Subject: IVGCVSW-4553 Fix Wrapper Object Use after free in UnitTests.hpp
X-Git-Tag: submit/tizen/20200730.023729~171
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=25334cf3d53fe7fff98776b44a199ca341f62f1a;p=platform%2Fupstream%2Farmnn.git

IVGCVSW-4553 Fix Wrapper Object Use after free in UnitTests.hpp

Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I174f83c9a454b5499f817baf0bddcd4a49cac523
---

diff --git a/src/armnn/test/UnitTests.hpp b/src/armnn/test/UnitTests.hpp
index 60d191f97..fbba9a247 100644
--- a/src/armnn/test/UnitTests.hpp
+++ b/src/armnn/test/UnitTests.hpp
@@ -70,6 +70,8 @@ void RunTestFunction(const char* testName, TFuncPtr testFunction, Args... args)
 
     auto testResult = (*testFunction)(workloadFactory, memoryManager, args...);
     CompareTestResultIfSupported(testName, testResult);
+
+    armnn::ProfilerManager::GetInstance().RegisterProfiler(nullptr);
 }
 
 #define ARMNN_SIMPLE_TEST_CASE(TestName, TestFunction) \