Skip to content

Commit 8b012a0

Browse files
authored
Arm run.sh fixes (#6043)
Fixes for vela compiler version, codegen flags and non-semihosted error behaviour
1 parent f4bbf19 commit 8b012a0

File tree

3 files changed

+9
-2
lines changed

3 files changed

+9
-2
lines changed

examples/arm/aot_arm_compiler.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -214,7 +214,12 @@ def forward(self, x):
214214
edge = edge.to_backend(
215215
ArmPartitioner(
216216
ArmCompileSpecBuilder()
217-
.ethosu_compile_spec("ethos-u55-128")
217+
.ethosu_compile_spec(
218+
"ethos-u55-128",
219+
system_config="Ethos_U55_High_End_Embedded",
220+
memory_mode="Shared_Sram",
221+
extra_flags="--debug-force-regor --output-format=raw",
222+
)
218223
.set_permute_memory_format(
219224
args.model_name in MODEL_NAME_TO_MODEL.keys()
220225
)

examples/arm/executor_runner/arm_executor_runner.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,10 +105,12 @@ Result<BufferCleanup> prepare_input_tensors(
105105
size_t num_inputs = method_meta.num_inputs();
106106
size_t num_allocated = 0;
107107

108+
#ifdef SEMIHOSTING
108109
ET_CHECK_OR_RETURN_ERROR(
109110
input_buffers.size() > 0 && num_inputs == input_buffers.size(),
110111
InvalidArgument,
111112
"Wrong number of inputs allocated compared to method");
113+
#endif
112114

113115
void** inputs =
114116
static_cast<void**>(allocator.allocate(num_inputs * sizeof(void*)));

examples/arm/setup.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ function setup_vela() {
216216
if [[ ! -e ethos-u-vela ]]; then
217217
git clone https://review.mlplatform.org/ml/ethos-u/ethos-u-vela
218218
repo_dir="${root_dir}/ethos-u-vela"
219-
base_rev=d362f5443f67b1e6213a9d8f124edff758efac96
219+
base_rev=57ce18c89ccc6f6309333dccb24ed30dc68b571f
220220
patch_repo
221221
fi
222222
cd "${root_dir}/ethos-u-vela"

0 commit comments

Comments
 (0)