commit 8df12f37531d57a10cba2f8b2e8b6a9065202dd5
author Isabella Gottardi <isabella.gottardi@arm.com> Wed Apr 07 17:15:31 2021 +0100
committer Alexander Efremov <alexander.efremov@arm.com> Mon Apr 12 14:00:49 2021 +0000
tree ba833d15649c3b0f885d57b40d3916970b3fd2c8
parent 37ce22ebc9cf3e8529d9914c0eed0f718243d961

MLECO-1870: Cherry pick profiling changes from dev to open source repo
* Documentation update

Change-Id: If85e7ebc44498840b291c408f14e66a5a5faa424
Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>

tests/use_case/img_class/ImgClassificationUCTest.cc

diff --git a/tests/use_case/img_class/ImgClassificationUCTest.cc b/tests/use_case/img_class/ImgClassificationUCTest.cc
index abfcc44..b989415 100644
--- a/tests/use_case/img_class/ImgClassificationUCTest.cc
+++ b/tests/use_case/img_class/ImgClassificationUCTest.cc
@@ -53,7 +53,7 @@
     hal_platform_init(&platform);
 
     /* Model wrapper object. */
-    arm::app::MobileNetModel model;    
+    arm::app::MobileNetModel model;
 
     /* Load the model. */
     REQUIRE(model.Init());
@@ -61,6 +61,8 @@
     /* Instantiate application context. */
     arm::app::ApplicationContext caseContext;
 
+    arm::app::Profiler profiler{&platform, "img_class"};
+    caseContext.Set<arm::app::Profiler&>("profiler", profiler);
     caseContext.Set<hal_platform&>("platform", platform);
     caseContext.Set<arm::app::Model&>("model", model);
     caseContext.Set<uint32_t>("imgIndex", 0);
@@ -99,6 +101,8 @@
     /* Instantiate application context. */
     arm::app::ApplicationContext caseContext;
 
+    arm::app::Profiler profiler{&platform, "img_class"};
+    caseContext.Set<arm::app::Profiler&>("profiler", profiler);
     caseContext.Set<hal_platform&>("platform", platform);
     caseContext.Set<arm::app::Model&>("model", model);
     caseContext.Set<uint32_t>("imgIndex", 0);