Skip to content

Commit 7a2a300

Browse files
authored
Arm backend: Fix flaky test LSTM by bumping allowed atol/rtol slightly (#9452)
This fix some rare fails on the LSTM test as we seem to have been too close with the values. Signed-off-by: Zingo Andersen <[email protected]>
1 parent 45219f3 commit 7a2a300

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/arm/test/test_arm_baremetal.sh

+2-2
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)