Skip to content

Commit 2d006c7

Browse files
mergennachinfacebook-github-bot
authored andcommitted
Arm backend: Refactor compile spec handling (Try 2)
Summary: Redoing #14111 with additional fixes Differential Revision: D82171193
1 parent de30390 commit 2d006c7

35 files changed

+745
-657
lines changed

backends/arm/TARGETS

Lines changed: 43 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,9 @@ python_library(
1414
"ethosu/partitioner.py"
1515
],
1616
deps = [
17-
":arm_partitioner",
17+
":arm_vela",
18+
":ethosu_compile_spec",
19+
"//executorch/backends/arm/tosa:partitioner",
1820
]
1921
)
2022
python_library(
@@ -25,7 +27,8 @@ python_library(
2527
"vgf/partitioner.py"
2628
],
2729
deps = [
28-
":arm_partitioner",
30+
":vgf_compile_spec",
31+
"//executorch/backends/arm/tosa:partitioner",
2932
]
3033
)
3134
python_library(
@@ -50,25 +53,11 @@ python_library(
5053
"//executorch/exir:lib",
5154
],
5255
)
56+
5357
python_library(
54-
name = "arm_partitioner",
55-
srcs = [
56-
"tosa/backend.py",
57-
"tosa/partitioner.py",
58-
],
59-
deps = [
60-
":arm_backend",
61-
":constants",
62-
"//executorch/backends/arm/debug:schema",
63-
"//executorch/backends/arm/operator_support:operator_support",
64-
"//executorch/backends/arm/_passes:passes",
65-
"//executorch/exir:lib",
66-
],
67-
)
68-
python_library(
69-
name = "arm_backend",
58+
name = "arm_compile_spec",
7059
srcs = [
71-
"arm_backend.py",
60+
"common/arm_compile_spec.py",
7261
],
7362
deps = [
7463
"fbsource//third-party/pypi/flatbuffers:flatbuffers",
@@ -79,11 +68,46 @@ python_library(
7968
"fbsource//third-party/tosa_tools/v1.00/serialization_lib/python/tosa:tosa",
8069
":arm_vela",
8170
":process_node",
71+
"//executorch/exir/backend:compile_spec_schema",
8272
"//executorch/backends/arm/operators:lib",
8373
"//executorch/backends/arm/operators:node_visitor",
8474
"//executorch/backends/arm/_passes:passes",
8575
],
8676
)
77+
78+
python_library(
79+
name = "ethosu_compile_spec",
80+
srcs = [
81+
"ethosu/compile_spec.py",
82+
],
83+
deps = [
84+
":arm_compile_spec",
85+
"//executorch/backends/arm/tosa:specification",
86+
],
87+
)
88+
89+
python_library(
90+
name = "tosa_compile_spec",
91+
srcs = [
92+
"tosa/compile_spec.py",
93+
],
94+
deps = [
95+
":arm_compile_spec",
96+
"//executorch/backends/arm/tosa:specification",
97+
],
98+
)
99+
100+
python_library(
101+
name = "vgf_compile_spec",
102+
srcs = [
103+
"vgf/compile_spec.py",
104+
],
105+
deps = [
106+
":arm_compile_spec",
107+
"//executorch/backends/arm/tosa:specification",
108+
],
109+
)
110+
87111
python_library(
88112
name = "process_node",
89113
srcs = ["process_node.py"],

backends/arm/arm_backend.py

Lines changed: 0 additions & 245 deletions
This file was deleted.

0 commit comments

Comments
 (0)