diff options
Diffstat (limited to 'scripts')
| -rw-r--r-- | scripts/cmake/native-sources.cmake | 5 | ||||
| -rw-r--r-- | scripts/py/gen_test_data_cpp.py | 4 |
2 files changed, 7 insertions, 2 deletions
diff --git a/scripts/cmake/native-sources.cmake b/scripts/cmake/native-sources.cmake index 1b1431a..dc8f263 100644 --- a/scripts/cmake/native-sources.cmake +++ b/scripts/cmake/native-sources.cmake @@ -33,6 +33,11 @@ endif() set(TENSORFLOW_LITE_MICRO_PLATFORM_LIB_NAME "libtensorflow-microlite.a") set(TENSORFLOW_LITE_MICRO_FLAGS "-DTF_LITE_STATIC_MEMORY -DACTIVATION_BUF_SRAM_SZ=0") +if (DEFINED VERIFY_TEST_OUTPUT) + message(STATUS "Test output verification flag is: ${VERIFY_TEST_OUTPUT}") + set(PROFILING_OPT "${PROFILING_OPT} -DVERIFY_TEST_OUTPUT=${VERIFY_TEST_OUTPUT}") +endif () + set(CMAKE_C_FLAGS "${PROFILING_OPT} ${LOG_FLAG} ${TENSORFLOW_LITE_MICRO_FLAGS}" CACHE INTERNAL "") diff --git a/scripts/py/gen_test_data_cpp.py b/scripts/py/gen_test_data_cpp.py index 7cc5f11..ea4bd6f 100644 --- a/scripts/py/gen_test_data_cpp.py +++ b/scripts/py/gen_test_data_cpp.py @@ -137,7 +137,7 @@ def main(args): if ifm_size == -1: ifm_size = get_npy_vec_size(filename) elif ifm_size != get_npy_vec_size(filename): - raise Exeception(f"ifm size changed for index {idx}") + raise Exception(f"ifm size changed for index {idx}") # Save the fm cc file base_name = "ofm" + str(idx) @@ -149,7 +149,7 @@ def main(args): if ofm_size == -1: ofm_size = get_npy_vec_size(filename) elif ofm_size != get_npy_vec_size(filename): - raise Exeception(f"ofm size changed for index {idx}") + raise Exception(f"ofm size changed for index {idx}") common_cc_filepath = os.path.join(args.source_folder_path, common_cc_filename) write_hpp_file(header_filename, common_cc_filepath, args.license_template, |