Arm backend: Re-enable W2L U85 INT test (pytorch#17944)

cc @digantdesai @SS-JIA @freddan80 @per @zingo @oscarandersson8218
@mansnils @Sebastian-Larsson @robell

Signed-off-by: Baris Demir <baris.demir@arm.com>
Co-authored-by: Baris Demir <baris.demir@arm.com>

Author: 2 people

backends/arm/test/models/test_w2l_arm.py

@@ -24,6 +24,11 @@
 
 input_t = Tuple[torch.Tensor]  # Input x
 
+quant_test_data = {
+    "per_channel_quantization=true": True,
+    "per_channel_quantization=false": False,
+}
+
 
 def get_test_inputs(batch_size, num_features, input_frames):
     return (torch.randn(batch_size, num_features, input_frames),)
@@ -99,14 +104,15 @@ def test_w2l_u55_INT():
 
 @pytest.mark.slow
 @common.XfailIfNoCorstone320
-@pytest.mark.skip(reason="Intermittent timeout issue: MLETORCH-856")
-def test_w2l_u85_INT():
+@common.parametrize("per_channel_quantization", quant_test_data)
+def test_w2l_u85_INT(per_channel_quantization):
     pipeline = EthosU85PipelineINT[input_t](
-        TestW2L.create_model(),
+        TestW2L.create_model("power_spectrum"),
         TestW2L.model_example_inputs,
         aten_ops=[],
         exir_ops=[],
         use_to_edge_transform_and_lower=True,
+        per_channel_quantization=per_channel_quantization,
     )
     pipeline.run()
 

backends/arm/test/ops/test_ceil.py

@@ -87,6 +87,7 @@ def test_ceil_tosa_INT(test_data: input_t1):
         module.exir_op,
         atol=0.06,
         rtol=0.01,
+        frobenius_threshold=0.2,
     )
     pipeline.run()
 

backends/arm/test/pytest.ini

@@ -2,4 +2,5 @@
 addopts = --strict-markers
 markers =
     slow: Tests that take long time
+    flaky: Tests that are known to be flaky/intermittent
     tosa_ref_model: Tests that use TOSA reference model # Temporary!