From 8df12f37531d57a10cba2f8b2e8b6a9065202dd5 Mon Sep 17 00:00:00 2001 From: Isabella Gottardi Date: Wed, 7 Apr 2021 17:15:31 +0100 Subject: MLECO-1870: Cherry pick profiling changes from dev to open source repo * Documentation update Change-Id: If85e7ebc44498840b291c408f14e66a5a5faa424 Signed-off-by: Isabella Gottardi --- tests/use_case/img_class/ImgClassificationUCTest.cc | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to '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 @@ TEST_CASE("Inference by index", "[.]") 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 @@ TEST_CASE("Inference by index", "[.]") /* Instantiate application context. */ arm::app::ApplicationContext caseContext; + arm::app::Profiler profiler{&platform, "img_class"}; + caseContext.Set("profiler", profiler); caseContext.Set("platform", platform); caseContext.Set("model", model); caseContext.Set("imgIndex", 0); @@ -99,6 +101,8 @@ TEST_CASE("Inference run all images", "[.]") /* Instantiate application context. */ arm::app::ApplicationContext caseContext; + arm::app::Profiler profiler{&platform, "img_class"}; + caseContext.Set("profiler", profiler); caseContext.Set("platform", platform); caseContext.Set("model", model); caseContext.Set("imgIndex", 0); -- cgit v1.2.1