[ExecuTorch] Arm Ethos: Add Linear operator tests

Pull Request resolved: #8663

As title.
ghstack-source-id: 268685731
@exported-using-ghexport

Differential Revision: [D70018299](https://our.internmc.facebook.com/intern/diff/D70018299/)

Author: digantdesai

backends/arm/test/ops/test_linear.py

@@ -127,7 +127,7 @@ def forward(self, x):
     def _test_linear_tosa_MI_pipeline(
         self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
     ):
-        (
+        tester = (
             ArmTester(
                 module,
                 example_inputs=test_data,
@@ -141,13 +141,14 @@ def _test_linear_tosa_MI_pipeline(
             .to_edge_transform_and_lower()
             .check_count({"torch.ops.higher_order.executorch_call_delegate": 1})
             .to_executorch()
-            .run_method_and_compare_outputs(inputs=test_data)
         )
+        if conftest.is_option_enabled("tosa_ref_model"):
+            tester.run_method_and_compare_outputs(inputs=test_data)
 
     def _test_linear_tosa_BI_pipeline(
         self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
     ):
-        (
+        tester = (
             ArmTester(
                 module,
                 example_inputs=test_data,
@@ -162,8 +163,9 @@ def _test_linear_tosa_BI_pipeline(
             .to_edge_transform_and_lower()
             .check_count({"torch.ops.higher_order.executorch_call_delegate": 1})
             .to_executorch()
-            .run_method_and_compare_outputs(inputs=test_data, qtol=1)
         )
+        if conftest.is_option_enabled("tosa_ref_model"):
+            tester.run_method_and_compare_outputs(inputs=test_data, qtol=1)
 
     def _test_linear_tosa_ethosu_BI_pipeline(
         self,
@@ -186,9 +188,11 @@ def _test_linear_tosa_ethosu_BI_pipeline(
             .to_executorch()
             .serialize()
         )
+        # TODO: Add FVP testing support.
         return tester
 
     @parameterized.expand(test_data_suite_rank1 + test_data_suite_rank4)
+    @pytest.mark.tosa_ref_model
     def test_linear_tosa_MI(
         self,
         test_name: str,
@@ -208,6 +212,7 @@ def test_linear_tosa_MI(
         )
 
     @parameterized.expand(test_data_suite_rank1 + test_data_suite_rank4)
+    @pytest.mark.tosa_ref_model
     def test_linear_tosa_BI(
         self,
         test_name: str,
@@ -249,6 +254,7 @@ def test_linear_tosa_u55_BI(
             tester.run_method_and_compare_outputs(qtol=1, inputs=test_data)
 
     @parameterized.expand(test_data_suite_rank1 + test_data_suite_rank4)
+    @pytest.mark.corstone_fvp
     def test_linear_tosa_u85_BI(
         self,
         test_name: str,

backends/arm/test/targets.bzl

@@ -4,11 +4,16 @@ load("@bazel_skylib//lib:paths.bzl", "paths")
 
 def define_arm_tests():
     # TODO Add more tests
-    test_files = native.glob(["passes/test_*.py"])
+    test_files = []
 
+    # Passes
+    test_files += native.glob(["passes/test_*.py"])
     # https://github.com/pytorch/executorch/issues/8606
     test_files.remove("passes/test_ioquantization_pass.py")
 
+    # Operators
+    test_files += native.glob(["ops/test_linear.py"])
+
     TESTS = {}
 
     for test_file in test_files:
@@ -26,8 +31,8 @@ def define_arm_tests():
                 "//executorch/kernels/quantized:custom_ops_generated_lib",
             ],
             deps = [
-                ":arm_tester",
-                ":conftest",
+                "//executorch/backends/arm/test:arm_tester",
+                "//executorch/backends/arm/test:conftest",
                 "//executorch/exir:lib",
                 "fbsource//third-party/pypi/pytest:pytest",
                 "fbsource//third-party/pypi/parameterized:parameterized",