diff options
Diffstat (limited to 'applications/baremetal')
-rw-r--r-- | applications/baremetal/CMakeLists.txt | 10 | ||||
-rw-r--r-- | applications/baremetal/main.cpp | 4 |
2 files changed, 11 insertions, 3 deletions
diff --git a/applications/baremetal/CMakeLists.txt b/applications/baremetal/CMakeLists.txt index 129fd3f..c9d965b 100644 --- a/applications/baremetal/CMakeLists.txt +++ b/applications/baremetal/CMakeLists.txt @@ -16,6 +16,14 @@ # limitations under the License. # +set(BAREMETAL_PATH "" CACHE PATH "Path to input, output and network for baremetal") + +if (IS_DIRECTORY ${BAREMETAL_PATH}) + ethosu_add_executable_test(baremetal_custom PRIVATE + SOURCES main.cpp) + target_include_directories(baremetal_custom PRIVATE ${BAREMETAL_PATH}) +endif() + ethosu_add_executable_test(baremetal_keyword_cnn PRIVATE SOURCES main.cpp) target_include_directories(baremetal_keyword_cnn PRIVATE models/keyword_spotting_cnn_small_int8) @@ -26,4 +34,4 @@ target_include_directories(baremetal_keyword_dnn PRIVATE models/keyword_spotting ethosu_add_executable_test(baremetal_keyword_ds_dnn PRIVATE SOURCES main.cpp) -target_include_directories(baremetal_keyword_ds_dnn PRIVATE models/keyword_spotting_ds_dnn_large_clustered_int8)
\ No newline at end of file +target_include_directories(baremetal_keyword_ds_dnn PRIVATE models/keyword_spotting_ds_dnn_large_clustered_int8) diff --git a/applications/baremetal/main.cpp b/applications/baremetal/main.cpp index e1e01de..4c0bf50 100644 --- a/applications/baremetal/main.cpp +++ b/applications/baremetal/main.cpp @@ -47,7 +47,7 @@ __attribute__((section(".bss.tensor_arena"), aligned(16))) uint8_t TFLuTensorAre InferenceProcess::InferenceProcess inferenceProcess(TFLuTensorArena, TENSOR_ARENA_SIZE); -uint8_t outputData[176] __attribute__((aligned(16), section("output_data_sec"))); +uint8_t outputData[sizeof(expectedOutputData)] __attribute__((aligned(16), section("output_data_sec"))); int runInference() { // Load inference data @@ -81,4 +81,4 @@ int runInference() { int main() { int ret = runInference(); return ret; -}
\ No newline at end of file +} |