From 7f6c0d09b4e63237a685674c74ec8c00d9b1a901 Mon Sep 17 00:00:00 2001 From: Oscar Andersson Date: Wed, 25 Jun 2025 10:32:12 +0200 Subject: [PATCH] Arm backend: Update flaky tests Signed-off-by: Oscar Andersson Change-Id: I71b302aa3f700af6e890c314623aee6d3f05bf6f --- backends/arm/test/models/test_deit_tiny_arm.py | 2 +- backends/arm/test/ops/test_amax.py | 2 +- backends/arm/test/ops/test_amin.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/backends/arm/test/models/test_deit_tiny_arm.py b/backends/arm/test/models/test_deit_tiny_arm.py index 38e631f70f0..a637db65dfd 100644 --- a/backends/arm/test/models/test_deit_tiny_arm.py +++ b/backends/arm/test/models/test_deit_tiny_arm.py @@ -52,7 +52,7 @@ def test_deit_tiny_tosa_BI(): aten_op=[], exir_op=[], use_to_edge_transform_and_lower=True, - atol=1, + atol=1.5, qtol=1, ) pipeline.run() diff --git a/backends/arm/test/ops/test_amax.py b/backends/arm/test/ops/test_amax.py index 0d1f4257b7b..bde9174de0f 100644 --- a/backends/arm/test/ops/test_amax.py +++ b/backends/arm/test/ops/test_amax.py @@ -97,7 +97,7 @@ def test_amax_u55_BI_not_delegated(): fvp_xfails = {"rank_4_mult_batches": "MLETORCH-517 : Multiple batches not supported"} -@common.parametrize("test_data", Amax.test_data, fvp_xfails) +@common.parametrize("test_data", Amax.test_data, fvp_xfails, strict=False) @common.XfailIfNoCorstone320 def test_amax_u85_BI(test_data: Amax.input_t): data, dim, keep_dims = test_data() diff --git a/backends/arm/test/ops/test_amin.py b/backends/arm/test/ops/test_amin.py index d83a5ee8839..89c4b71e5af 100644 --- a/backends/arm/test/ops/test_amin.py +++ b/backends/arm/test/ops/test_amin.py @@ -106,7 +106,7 @@ def test_amin_u55_BI_not_delegated(): fvp_xfails = {"rank_4_mult_batches": "MLETORCH-517 : Multiple batches not supported"} -@common.parametrize("test_data", Amin.test_data, fvp_xfails) +@common.parametrize("test_data", Amin.test_data, fvp_xfails, strict=False) @common.XfailIfNoCorstone320 def test_amin_u85_BI(test_data: Amin.input_t): data, dim, keep_dims = test_data()