From d15ee15ebe7b3a338385b2c948f092fe806f5d38 Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Wed, 18 Mar 2020 12:51:43 +0000 Subject: IVGCVSW-4552 Fix Wrapper Object Use after free in ProfilerTests Signed-off-by: Finn Williams Change-Id: I84ab68c048675a39a67045430b2dde571b45e479 --- src/armnn/test/ProfilerTests.cpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src') diff --git a/src/armnn/test/ProfilerTests.cpp b/src/armnn/test/ProfilerTests.cpp index 9376fa4cea..40e9f470ad 100644 --- a/src/armnn/test/ProfilerTests.cpp +++ b/src/armnn/test/ProfilerTests.cpp @@ -57,6 +57,8 @@ void RegisterUnregisterProfilerSingleThreadImpl(bool &res) // Check that the profiler has been un-registered for this thread. res &= !profilerManager.GetProfiler(); + + armnn::ProfilerManager::GetInstance().RegisterProfiler(nullptr); } } // namespace @@ -351,6 +353,7 @@ BOOST_AUTO_TEST_CASE(ProfilerJsonPrinter) "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(); -- cgit v1.2.1