From 3bd342372638b185d097308445747a1a668fce01 Mon Sep 17 00:00:00 2001 From: Kristofer Jonsson Date: Mon, 30 Aug 2021 13:55:55 +0200 Subject: Exporting define for layer by layer profiler Libraries that link the layer_by_layer_profiler interface library will also get the LAYER_BY_LAYER_PROFILE define appending to the build arguments. This allows the preprocessor to select which profiler to use. Change-Id: I85fec66d2dd9949fd7fe395fb44e49bda1539925 --- applications/inference_process/src/inference_process.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'applications') diff --git a/applications/inference_process/src/inference_process.cpp b/applications/inference_process/src/inference_process.cpp index 2a0b687..09439f8 100644 --- a/applications/inference_process/src/inference_process.cpp +++ b/applications/inference_process/src/inference_process.cpp @@ -24,7 +24,7 @@ #include "tensorflow/lite/schema/schema_generated.h" #include "arm_profiler.hpp" -#ifdef ETHOSU +#ifdef LAYER_BY_LAYER_PROFILER #include "layer_by_layer_profiler.hpp" #endif #include "ethosu_log.h" @@ -204,7 +204,7 @@ bool InferenceProcess::runJob(InferenceJob &job) { // Create the TFL micro interpreter tflite::AllOpsResolver resolver; -#ifdef ETHOSU +#ifdef LAYER_BY_LAYER_PROFILER tflite::LayerByLayerProfiler profiler; #else tflite::ArmProfiler profiler; -- cgit v1.2.1