From 6006f680a9f22270cd5c20f7531b057cc20902c4 Mon Sep 17 00:00:00 2001
From: Digant Desai <digantdesai@meta.com>
Date: Mon, 17 Mar 2025 10:28:21 -0700
Subject: [PATCH] [ExecuTorch] Arm Ethos: Add Linear operator tests

As title.

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

[ghstack-poisoned]
---
 backends/arm/test/TARGETS     | 13 ++++---------
 backends/arm/test/targets.bzl | 11 ++++++++---
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/backends/arm/test/TARGETS b/backends/arm/test/TARGETS
index 58239913e29..38a4ec18787 100644
--- a/backends/arm/test/TARGETS
+++ b/backends/arm/test/TARGETS
@@ -19,7 +19,6 @@ python_library(
     srcs = ["runner_utils.py"],
     deps = [
         ":conftest",
-        "//executorch/backends/xnnpack/test/tester:tester",
         "//executorch/backends/arm:arm_backend",
         "//executorch/exir:lib",
         "//executorch/exir/backend:compile_spec_schema",
@@ -31,10 +30,7 @@ python_library(
     srcs = ["common.py"],
     deps = [
         ":runner_utils",
-        "//executorch/backends/xnnpack/test/tester:tester",
-        "//executorch/backends/arm:arm_backend",
-        "//executorch/exir:lib",
-        "//executorch/exir/backend:compile_spec_schema",
+        "//executorch/backends/arm:tosa_specification",
         "fbsource//third-party/pypi/pytest:pytest",
     ]
 )
@@ -43,12 +39,11 @@ python_library(
     name = "arm_tester",
     srcs = glob(["tester/*.py"]),
     deps = [
-        ":runner_utils",
         ":common",
-        "//executorch/backends/arm:tosa_mapping",
-        "//executorch/backends/arm:tosa_specification",
-        "//executorch/backends/arm/quantizer:arm_quantizer",
+        "//executorch/backends/xnnpack/test/tester:tester",
         "//executorch/backends/arm:arm_partitioner",
+        "//executorch/backends/arm/quantizer:arm_quantizer",
+        "//executorch/backends/arm:tosa_mapping",
         "//executorch/devtools/backend_debug:delegation_info",
         "fbsource//third-party/pypi/tabulate:tabulate",
     ]
diff --git a/backends/arm/test/targets.bzl b/backends/arm/test/targets.bzl
index 3335dc958cd..e97b46cb977 100644
--- a/backends/arm/test/targets.bzl
+++ b/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",