From 22fb06cb50228d7fe4b01efe364fdb93d25a0ba1 Mon Sep 17 00:00:00 2001 From: Erik Lundell Date: Tue, 23 Jul 2024 14:57:09 +0200 Subject: [PATCH] Add flakyness mark to conv BI test Signed-off-by: Erik Lundell Change-Id: I391003f8480283872fdc0566e489bb9bb3926c6f --- backends/arm/test/ops/test_depthwise_conv.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/backends/arm/test/ops/test_depthwise_conv.py b/backends/arm/test/ops/test_depthwise_conv.py index 7eacbac432b..8389b423e5e 100644 --- a/backends/arm/test/ops/test_depthwise_conv.py +++ b/backends/arm/test/ops/test_depthwise_conv.py @@ -9,6 +9,8 @@ from typing import Tuple +import pytest + import torch from executorch.backends.arm.test import common from executorch.backends.arm.test.ops.test_conv import Conv2d @@ -189,7 +191,9 @@ def _test_dw_conv2d_u55_BI_pipeline( def test_dw_conv2d_tosa_MI(self, test_name, model): self._test_dw_conv2d_tosa_MI_pipeline(model, model.get_inputs()) + # TODO: Investigate flakyness (MLTORCH-307) @parameterized.expand(testsuite) + @pytest.mark.flaky(reruns=3) def test_dw_conv2d_tosa_BI(self, test_name, model): self._test_dw_conv2d_tosa_BI_pipeline(model, model.get_inputs())