Skip to content

Commit 47a5c97

Browse files
committed
Arm backend: Fix flaky test LSTM by bumping allowed atol/rtol slightly
Signed-off-by: Zingo Andersen <[email protected]> Change-Id: I58fec922a83f507ad0229680a232c8c8b16523eb
1 parent 6fb42ef commit 47a5c97

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/arm/test/test_arm_baremetal.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,13 +130,13 @@ test_models_ethosu_fvp() { # End to End model tests using model_test.py
130130
echo "${TEST_SUITE_NAME}: Test ethos-u target Ethos-U55"
131131
python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u55-128 --model=mv2 --extra_flags="-DET_ATOL=2.00 -DET_RTOL=2.00"
132132
python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u55-64 --model=mv3 --extra_flags="-DET_ATOL=5.00 -DET_RTOL=5.00"
133-
python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u55-256 --model=lstm --extra_flags="-DET_ATOL=0.02 -DET_RTOL=0.02"
133+
python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u55-256 --model=lstm --extra_flags="-DET_ATOL=0.03 -DET_RTOL=0.03"
134134

135135
# Ethos-U85
136136
echo "${TEST_SUITE_NAME}: Test ethos-u target Ethos-U85"
137137
python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u85-256 --model=mv2 --extra_flags="-DET_ATOL=2.00 -DET_RTOL=2.00"
138138
python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u85-1024 --model=mv3 --extra_flags="-DET_ATOL=5.00 -DET_RTOL=5.00"
139-
python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u85-128 --model=lstm --extra_flags="-DET_ATOL=0.02 -DET_RTOL=0.02"
139+
python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u85-128 --model=lstm --extra_flags="-DET_ATOL=0.03 -DET_RTOL=0.03"
140140
echo "${TEST_SUITE_NAME}: PASS"
141141
}
142142

0 commit comments

Comments
 (0)