commit d15ee15ebe7b3a338385b2c948f092fe806f5d38
author Finn Williams <Finn.Williams@arm.com> Wed Mar 18 12:51:43 2020 +0000
committer finn.williams <finn.williams@arm.com> Thu Mar 19 10:18:00 2020 +0000
tree d45b9924e97e10f199e65fdc4b83611fbc67a69c
parent a83af7b388a1ac721c6a85b0e80ac2950687dfad

IVGCVSW-4552 Fix Wrapper Object Use after free in ProfilerTests

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

src/armnn/test/ProfilerTests.cpp

diff --git a/src/armnn/test/ProfilerTests.cpp b/src/armnn/test/ProfilerTests.cpp
index 9376fa4..40e9f47 100644
--- a/src/armnn/test/ProfilerTests.cpp
+++ b/src/armnn/test/ProfilerTests.cpp
@@ -57,6 +57,8 @@
 
     // Check that the profiler has been un-registered for this thread.
     res &= !profilerManager.GetProfiler();
+
+    armnn::ProfilerManager::GetInstance().RegisterProfiler(nullptr);
 }
 
 } // namespace
@@ -351,6 +353,7 @@
                               "unit\": \"us\"\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n}\n");
 
     BOOST_CHECK(output == blessedOutput);
+    armnn::ProfilerManager::GetInstance().RegisterProfiler(nullptr);
 }
 
 BOOST_AUTO_TEST_SUITE_END();