diff options
| author | Davide Grohmann <davide.grohmann@arm.com> | 2022-08-08 17:12:08 +0200 |
|---|---|---|
| committer | Davide Grohmann <davide.grohmann@arm.com> | 2022-08-17 09:56:14 +0200 |
| commit | 5508acb0c7803769b9bbda08762b232c13272bd9 (patch) | |
| tree | cf62375a3d12e59c821318a13abaa510792fff69 /scripts | |
| parent | 06f954bb6dd56921f1fb38e3e36405e3f1c70ef9 (diff) | |
| download | ethos-u-core-platform-5508acb0c7803769b9bbda08762b232c13272bd9.tar.gz | |
Generalize test models to be reused
Change-Id: I47705cfde2788fa74f8bbcd07100188e4690dbf2
Diffstat (limited to 'scripts')
| -rwxr-xr-x | scripts/run_platform.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/scripts/run_platform.py b/scripts/run_platform.py index a39de8b..0629fef 100755 --- a/scripts/run_platform.py +++ b/scripts/run_platform.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 # -# Copyright (c) 2021 Arm Limited. All rights reserved. +# Copyright (c) 2021-2022 Arm Limited. # # SPDX-License-Identifier: Apache-2.0 # @@ -48,7 +48,7 @@ def ta_parse_raw(ta_raw): if (index > 1): raise Exception("Illegal index value - Should be '0' or '1'") (ta_parsed)[index] = value - + return ta_parsed def build_core_platform(output_folder, target, toolchain, memory_model, memory_arena, pmu, @@ -183,7 +183,12 @@ def optimize_network(output_folder, network_path, accelerator_conf): res = run_cmd(vela_cmd) optimized_model_path = output_folder/(network_path.stem + "_vela.tflite") model_name = network_path.stem - dump_c_header(optimized_model_path, output_folder/"model.h", "networkModelData", "network_model_sec", 16, extra_data=f"const char *modelName=\"{model_name}\";\n") + dump_c_header(optimized_model_path, output_folder/"model.h", "networkModelData", "network_model_sec", 16, extra_data=f""" +#include <stddef.h> + +const size_t tensorArenaSize = 2000000; +const char* modelName = \"{model_name}\"; +""") def run_model(output_folder): build_folder = output_folder/"model"/"build" |