Arm backend: Refactor compile spec handling (Try 2) (pytorch#14191)

Summary:
Pull Request resolved: pytorch#14191

Redoing pytorch#14111 with additional fixes

Differential Revision: D82171193

Author: mergennachin

backends/arm/TARGETS

@@ -6,28 +6,6 @@
 # @noautodeps
 load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
 
-python_library(
-    name = "ethosu_partitioner",
-    srcs = [
-        "ethosu/__init__.py",
-        "ethosu/backend.py",
-        "ethosu/partitioner.py"
-    ],
-    deps = [
-        ":arm_partitioner",
-    ]
-)
-python_library(
-    name = "vgf_partitioner",
-    srcs = [
-        "vgf/__init__.py",
-        "vgf/backend.py",
-        "vgf/partitioner.py"
-    ],
-    deps = [
-        ":arm_partitioner",
-    ]
-)
 python_library(
     name = "constants",
     srcs = [
@@ -50,25 +28,11 @@ python_library(
         "//executorch/exir:lib",
     ],
 )
+
 python_library(
-    name = "arm_partitioner",
-    srcs = [
-        "tosa/backend.py",
-        "tosa/partitioner.py",
-    ],
-    deps = [
-        ":arm_backend",
-        ":constants",
-        "//executorch/backends/arm/debug:schema",
-        "//executorch/backends/arm/operator_support:operator_support",
-        "//executorch/backends/arm/_passes:passes",
-        "//executorch/exir:lib",
-    ],
-)
-python_library(
-    name = "arm_backend",
+    name = "arm_compile_spec",
     srcs = [
-        "arm_backend.py",
+        "common/arm_compile_spec.py",
     ],
     deps = [
         "fbsource//third-party/pypi/flatbuffers:flatbuffers",
@@ -79,11 +43,56 @@ python_library(
         "fbsource//third-party/tosa_tools/v1.00/serialization_lib/python/tosa:tosa",
         ":arm_vela",
         ":process_node",
+        "//executorch/exir/backend:compile_spec_schema",
         "//executorch/backends/arm/operators:lib",
         "//executorch/backends/arm/operators:node_visitor",
         "//executorch/backends/arm/_passes:passes",
     ],
 )
+
+python_library(
+    name = "ethosu",
+    srcs = [
+        "ethosu/__init__.py",
+        "ethosu/backend.py",
+        "ethosu/compile_spec.py",
+        "ethosu/partitioner.py",
+    ],
+    deps = [
+        ":arm_compile_spec",
+        ":arm_vela",
+        "//executorch/backends/arm/tosa:specification",
+        "//executorch/backends/arm/tosa:partitioner",
+    ],
+)
+
+python_library(
+    name = "tosa_compile_spec",
+    srcs = [
+        "tosa/__init__.py",
+        "tosa/compile_spec.py",
+    ],
+    deps = [
+        ":arm_compile_spec",
+        "//executorch/backends/arm/tosa:specification",
+    ],
+)
+
+python_library(
+    name = "vgf",
+    srcs = [
+        "vgf/__init__.py",
+        "vgf/backend.py",
+        "vgf/compile_spec.py",
+        "vgf/partitioner.py",
+    ],
+    deps = [
+        ":arm_compile_spec",
+        "//executorch/backends/arm/tosa:specification",
+        "//executorch/backends/arm/tosa:partitioner",
+    ],
+)
+
 python_library(
     name = "process_node",
     srcs = ["process_node.py"],