From c9ed74ba52d1db7cb33e885a21dd12ea81aa79a8 Mon Sep 17 00:00:00 2001 From: Fang-Ching Date: Tue, 6 May 2025 17:31:03 +0100 Subject: [PATCH] Arm backend: increase atol to tosa_BI test in test_conv_constant_pad_nd.py The output verification occasionally fails in the tosa_BI test due to small numerical differences. Increased atol to reduce flakiness. Signed-off-by: Fang-Ching Change-Id: I746740e088346ac8008f3f2a43008a5c6a27cb93 --- backends/arm/test/ops/test_conv_constant_pad_nd.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/backends/arm/test/ops/test_conv_constant_pad_nd.py b/backends/arm/test/ops/test_conv_constant_pad_nd.py index 026b5d1dc4d..61497578fb6 100644 --- a/backends/arm/test/ops/test_conv_constant_pad_nd.py +++ b/backends/arm/test/ops/test_conv_constant_pad_nd.py @@ -110,5 +110,7 @@ def test_constant_pad_nd_tosa_BI(test_data: Tuple): (test_data,), aten_op, exir_op, + atol=0.005, # TODO: Investigate flakyness (MLETORCH-989) + rtol=0.01, ) pipeline.run()