From b444f6614823e97a2a4aa2f943f4960fee6cf98c Mon Sep 17 00:00:00 2001 From: Saoirse Stewart Date: Mon, 12 May 2025 13:15:38 +0100 Subject: [PATCH] Arm Backend: Updated test_var to use qtol=1 Signed-off-by: Saoirse Stewart Change-Id: I9ef52fc0fe493bf5a7e03cb85236b8d81f00d81b --- backends/arm/test/ops/test_conv_combos.py | 10 +--------- backends/arm/test/tester/test_pipeline.py | 2 +- 2 files changed, 2 insertions(+), 10 deletions(-) diff --git a/backends/arm/test/ops/test_conv_combos.py b/backends/arm/test/ops/test_conv_combos.py index 7f54fa226aa..c06a6e666ec 100644 --- a/backends/arm/test/ops/test_conv_combos.py +++ b/backends/arm/test/ops/test_conv_combos.py @@ -430,16 +430,8 @@ def test_convolution_2d_tosa_MI_avgpool2d(test_data: torch.Tensor): pipeline.run() -x_fails = { - "combo_conv_avgpool_20_x_4d": "AssertionError: Output 0 does not match reference output.", - "combo_conv_avgpool_4d": "AssertionError: Output 0 does not match reference output.", - "combo_conv_avgpool_5_x_4d_randn": "AssertionError: Output 0 does not match reference output.", - "combo_conv_avgpool_2_x_4d": "AssertionError: Output 0 does not match reference output.", -} - - @pytest.mark.flaky(reruns=5) # TODO: Investigate flakyness (MLTORCH-307) -@common.parametrize("test_data", ComboConvAvgPool2d.test_data, x_fails) +@common.parametrize("test_data", ComboConvAvgPool2d.test_data) def test_convolution_2d_tosa_BI_avgpool2d(test_data: torch.Tensor): model = ComboConvAvgPool2d() pipeline = TosaPipelineBI[input_t1]( diff --git a/backends/arm/test/tester/test_pipeline.py b/backends/arm/test/tester/test_pipeline.py index 58c7c657250..58f8bda5391 100644 --- a/backends/arm/test/tester/test_pipeline.py +++ b/backends/arm/test/tester/test_pipeline.py @@ -282,7 +282,7 @@ def __init__( custom_path: str = None, atol: float = 1e-03, rtol: float = 1e-03, - qtol: int = 0, + qtol: int = 1, ): tosa_profiles = { "0.80": TosaSpecification.create_from_string("TOSA-0.80+BI"),