@@ -17,6 +17,7 @@ _setup_msg="please refer to ${et_root_dir}/examples/arm/setup.sh to properly ins
17
17
18
18
19
19
TEST_SUITE=$1
20
+ TOSA_VERSION=" ${2:- TOSA-1.0+INT} "
20
21
21
22
# Source the tools
22
23
# This should be prepared by the setup.sh
@@ -138,8 +139,8 @@ test_run_ethosu_fvp() { # End to End model tests using run.sh
138
139
139
140
# TOSA quantized
140
141
echo " ${TEST_SUITE_NAME} : Test ethos-u target TOSA"
141
- examples/arm/run.sh --et_build_root=arm_test/test_run --target=TOSA --model_name=add
142
- examples/arm/run.sh --et_build_root=arm_test/test_run --target=TOSA --model_name=mul
142
+ examples/arm/run.sh --et_build_root=arm_test/test_run --target=${TOSA_VERSION} --model_name=add
143
+ examples/arm/run.sh --et_build_root=arm_test/test_run --target=${TOSA_VERSION} --model_name=mul
143
144
144
145
# Ethos-U55
145
146
echo " ${TEST_SUITE_NAME} : Test ethos-u target Ethos-U55"
@@ -168,17 +169,17 @@ test_models_tosa() { # End to End model tests using model_test.py
168
169
169
170
# TOSA quantized
170
171
echo " ${TEST_SUITE_NAME} : Test ethos-u target TOSA"
171
- python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=mv2
172
- python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=mv3
173
- python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=lstm
174
- python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=edsr
175
- # python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=emformer_transcribe # Takes long time to run
176
- # python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=emformer_join # Takes long time to run
177
- python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=w2l
178
- python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=ic3
179
- python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=ic4
180
- python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=resnet18
181
- python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=TOSA --model=resnet50
172
+ python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=mv2
173
+ python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=mv3
174
+ python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=lstm
175
+ python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=edsr
176
+ # python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=emformer_transcribe # Takes long time to run
177
+ # python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=emformer_join # Takes long time to run
178
+ python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=w2l
179
+ python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=ic3
180
+ python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=ic4
181
+ python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=resnet18
182
+ python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=${TOSA_VERSION} --model=resnet50
182
183
183
184
echo " ${TEST_SUITE_NAME} : PASS"
184
185
}
0 commit comments