Arm backend: Add U55 and U85 tests for deit_tiny (#16145)

Adds tests of deit_tiny for U55 and U85.

Signed-off-by: Oscar Andersson <oscar.andersson@arm.com>

Author: oscarandersson8218

backends/arm/test/models/test_deit_tiny_arm.py

@@ -3,8 +3,6 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-import logging
-
 from typing import Tuple
 
 import timm  # type: ignore[import-untyped]
@@ -14,6 +12,8 @@
 from executorch.backends.arm.test import common
 
 from executorch.backends.arm.test.tester.test_pipeline import (
+    EthosU55PipelineINT,
+    EthosU85PipelineINT,
     TosaPipelineFP,
     TosaPipelineINT,
     VgfPipeline,
@@ -25,10 +25,8 @@
 )
 from torchvision import transforms  # type: ignore[import-untyped]
 
-logger = logging.getLogger(__name__)
-
-
 deit_tiny = timm.models.deit.deit_tiny_patch16_224(pretrained=True)
+
 deit_tiny.eval()
 
 normalize = transforms.Normalize(
@@ -63,6 +61,37 @@ def test_deit_tiny_tosa_INT():
     pipeline.run()
 
 
+def test_deit_tiny_u55_INT():
+    pipeline = EthosU55PipelineINT[input_t](
+        deit_tiny,
+        model_inputs,
+        aten_ops=[],
+        exir_ops=[],
+        use_to_edge_transform_and_lower=True,
+        atol=1.5,
+        qtol=1,
+    )
+    # Multiple partitions
+    pipeline.pop_stage("check_count.exir")
+    # Don't run inference as model is too large for Corstone-300
+    pipeline.pop_stage("run_method_and_compare_outputs")
+    pipeline.run()
+
+
+@common.XfailIfNoCorstone320
+def test_deit_tiny_u85_INT():
+    pipeline = EthosU85PipelineINT[input_t](
+        deit_tiny,
+        model_inputs,
+        aten_ops=[],
+        exir_ops=[],
+        use_to_edge_transform_and_lower=True,
+        atol=1.5,
+        qtol=1,
+    )
+    pipeline.run()
+
+
 @common.SkipIfNoModelConverter
 def test_deit_tiny_vgf_INT():
     pipeline = VgfPipeline[input_t](