Skip to content

Commit e241eb0

Browse files
committed
Arm backends: New quantized FVP tests for cortex-M
tested using, ./examples/arm/run.sh --target=ethos-u85-256 --model_name=lstm --bundleio
1 parent 0c27535 commit e241eb0

File tree

2 files changed

+25
-2
lines changed

2 files changed

+25
-2
lines changed

backends/arm/test/test_arm_baremetal.sh

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,13 @@ test_run_ethosu_fvp() { # End to End model tests using run.sh
154154
echo "${TEST_SUITE_NAME}: Test ethos-u target Ethos-U85"
155155
examples/arm/run.sh --et_build_root=arm_test/test_run --target=ethos-u85-128 --model_name=add
156156
examples/arm/run.sh --et_build_root=arm_test/test_run --target=ethos-u85-128 --model_name=mul
157+
158+
# Cortex-M op tests
159+
examples/arm/run.sh --et_build_root=arm_test/test_run --target=ethos-u55-128 --model_name=qadd --bundleio
160+
examples/arm/run.sh --et_build_root=arm_test/test_run --target=ethos-u55-128 --model_name=qops --bundleio
161+
examples/arm/run.sh --et_build_root=arm_test/test_run --target=ethos-u55-128 --model_name=qops --bundleio --no_delegate --portable_kernels="aten::sub.out,aten::add.out,aten::mul.out"
162+
examples/arm/run.sh --et_build_root=arm_test/test_run --target=ethos-u85-128 --model_name=qops --bundleio
163+
157164
echo "${TEST_SUITE_NAME}: PASS"
158165
}
159166

examples/arm/run.sh

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,8 +177,24 @@ backends/arm/scripts/build_portable_kernels.sh --et_build_root="${et_build_root}
177177

178178
if [[ -z "$model_name" ]]; then
179179
# the test models run, and whether to delegate
180-
test_model=( "softmax" "add" "add3" "mv2" )
181-
model_compiler_flags=( "" "--delegate" "--delegate" "--delegate --quantize" )
180+
test_model=(
181+
"softmax" # 0
182+
"add" # 1
183+
"add3" # 2
184+
"qadd" # 3
185+
"qadd3" # 4
186+
"qops" # 5
187+
"mv2" # 6
188+
)
189+
model_compiler_flags=(
190+
"" # 0 softmax
191+
"--delegate" # 1 add
192+
"--delegate" # 2 add3
193+
"--delegate --quantize" # 3 qadd
194+
"--delegate --quantize" # 4 qadd3
195+
"--delegate --quantize" # 5 qops
196+
"--delegate --quantize" # 6 mv2
197+
)
182198
else
183199
test_model=( "$model_name" )
184200
model_compiler_flags=( "$aot_arm_compiler_flag_delegate $aot_arm_compiler_flag_quantize $aot_arm_compiler_flags" )

0 commit comments

Comments
 (0)