Arm backend: Unmark test_bmm:tosa_bi as flaky

Replace pytest.mark_flaky with qtol=1 for TOSA BI tests in test_bmm.py

Signed-off-by: Oscar Andersson <[email protected]>
Change-Id: Idb71178cb458557a74d1518b8c8c3283ff7ff9ad

Author: oscarandersson8218

backends/arm/test/ops/test_bmm.py

@@ -71,14 +71,14 @@ def test_bmm_tosa_MI_single_input(test_data: input_t1):
     pipeline.run()
 
 
-@pytest.mark.flaky(reruns=5)  # TODO: Investigate flakyness (MLETORCH-534)
 @common.parametrize("test_data", BMM.test_data_generators)
 def test_bmm_tosa_BI(test_data: input_t1):
-    pipeline = TosaPipelineBI[input_t1](BMM(), test_data(), aten_op_bmm, exir_op_bmm)
+    pipeline = TosaPipelineBI[input_t1](
+        BMM(), test_data(), aten_op_bmm, exir_op_bmm, qtol=1
+    )
     pipeline.run()
 
 
-@pytest.mark.flaky(reruns=5)  # TODO: Investigate flakyness (MLETORCH-534)
 @common.parametrize("test_data", BMMSingleInput.test_data_generators)
 def test_bmm_tosa_BI_single_input(test_data: input_t1):
     pipeline = TosaPipelineBI[input_t1](

backends/arm/test/ops/test_mm.py

@@ -41,7 +41,7 @@ def test_mm_tosa_MI(test_data: Tuple):
 
 @common.parametrize("test_data", MM.test_data_generators)
 def test_mm_tosa_BI(test_data: Tuple):
-    TosaPipelineBI[test_t](MM(), test_data(), MM.aten_op, MM.exir_op).run()
+    TosaPipelineBI[test_t](MM(), test_data(), MM.aten_op, MM.exir_op, qtol=1).run()
 
 
 @common.parametrize("test_data", MM.test_data_generators)