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

Summary:
Pull Request resolved: #14191

Redoing #14111 with additional fixes

Reviewed By: digantdesai

Differential Revision: D82171193

Author: mergennachin

backends/arm/TARGETS

@@ -6,29 +6,6 @@
 # @noautodeps
 load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
 
-runtime.python_library(
-    name = "ethosu_partitioner",
-    srcs = [
-        "ethosu/__init__.py",
-        "ethosu/backend.py",
-        "ethosu/partitioner.py"
-    ],
-    deps = [
-        ":arm_vela",
-        "//executorch/backends/arm/tosa:arm_partitioner",
-    ]
-)
-runtime.python_library(
-    name = "vgf_partitioner",
-    srcs = [
-        "vgf/__init__.py",
-        "vgf/backend.py",
-        "vgf/partitioner.py"
-    ],
-    deps = [
-        "//executorch/backends/arm/tosa:arm_partitioner",
-    ]
-)
 runtime.python_library(
     name = "constants",
     srcs = [
@@ -51,10 +28,11 @@ runtime.python_library(
         "//executorch/exir:lib",
     ],
 )
+
 runtime.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",
@@ -64,11 +42,55 @@ runtime.python_library(
         "fbsource//third-party/tosa_tools/v0.80/serialization_lib/python/tosa:tosa",
         "fbsource//third-party/tosa_tools/v1.00/serialization_lib/python/tosa:tosa",
         ":process_node",
+        "//executorch/exir/backend:compile_spec_schema",
         "//executorch/backends/arm/operators:lib",
         "//executorch/backends/arm/operators:node_visitor",
         "//executorch/backends/arm/_passes:passes",
     ],
 )
+runtime.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",
+    ],
+)
+
+runtime.python_library(
+    name = "tosa_compile_spec",
+    srcs = [
+        "tosa/__init__.py",
+        "tosa/compile_spec.py",
+    ],
+    deps = [
+        ":arm_compile_spec",
+        "//executorch/backends/arm/tosa:specification",
+    ],
+)
+
+runtime.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",
+    ],
+)
+
 runtime.python_library(
     name = "process_node",
     srcs = ["process_node.py"],