diff options
| author | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2021-08-30 13:55:55 +0200 |
|---|---|---|
| committer | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2021-08-30 13:55:55 +0200 |
| commit | 3bd342372638b185d097308445747a1a668fce01 (patch) | |
| tree | 753c156a18a6001db3e9d89f806bc26cdb3e7b40 | |
| parent | f40a3a9a120cb9510942ecf5aa2bd12a99de3516 (diff) | |
| download | ethos-u-core-software-3bd342372638b185d097308445747a1a668fce01.tar.gz | |
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
| -rw-r--r-- | applications/inference_process/src/inference_process.cpp | 4 | ||||
| -rw-r--r-- | lib/layer_by_layer_profiler/CMakeLists.txt | 17 |
2 files changed, 16 insertions, 5 deletions
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; diff --git a/lib/layer_by_layer_profiler/CMakeLists.txt b/lib/layer_by_layer_profiler/CMakeLists.txt index 7a16da2..1e3d012 100644 --- a/lib/layer_by_layer_profiler/CMakeLists.txt +++ b/lib/layer_by_layer_profiler/CMakeLists.txt @@ -21,6 +21,17 @@ if (NOT TARGET ethosu_core_driver) endif() add_library(layer_by_layer_profiler INTERFACE) -target_link_libraries(layer_by_layer_profiler INTERFACE ethosu_core_driver tflu event_recorder ethosu_log) -target_include_directories(layer_by_layer_profiler INTERFACE include) -target_sources(layer_by_layer_profiler INTERFACE src/layer_by_layer_profiler.cpp) + +target_link_libraries(layer_by_layer_profiler INTERFACE + ethosu_core_driver + tflu event_recorder + ethosu_log) + +target_include_directories(layer_by_layer_profiler INTERFACE + include) + +target_sources(layer_by_layer_profiler INTERFACE + src/layer_by_layer_profiler.cpp) + +target_compile_definitions(layer_by_layer_profiler INTERFACE + LAYER_BY_LAYER_PROFILER) |