File tree Expand file tree Collapse file tree 2 files changed +34
-7
lines changed Expand file tree Collapse file tree 2 files changed +34
-7
lines changed Original file line number Diff line number Diff line change 11load ("@fbsource//xplat/executorch/build:runtime_wrapper.bzl" , "runtime" )
22
33def define_common_targets ():
4+ runtime .python_library (
5+ name = "aoti_partitioner" ,
6+ srcs = [
7+ "aoti_partitioner.py" ,
8+ ],
9+ visibility = [
10+ "//executorch/..." ,
11+ ],
12+ deps = [
13+ "//caffe2:torch" ,
14+ "//executorch/exir/backend:partitioner" ,
15+ "//executorch/exir/backend:utils" ,
16+ ],
17+ )
18+
19+ runtime .python_library (
20+ name = "aoti_backend" ,
21+ srcs = [
22+ "aoti_backend.py" ,
23+ ],
24+ visibility = [
25+ "//executorch/..." ,
26+ ],
27+ deps = [
28+ "//caffe2:torch" ,
29+ "//executorch/backends/aoti/passes:passes" ,
30+ "//executorch/exir/_serialize:lib" ,
31+ "//executorch/exir/backend:backend_details" ,
32+ "//executorch/exir/backend:compile_spec_schema" ,
33+ ],
34+ )
35+
436 # AOTI common shims functionality
537 runtime .cxx_library (
638 name = "common_shims" ,
Original file line number Diff line number Diff line change @@ -11,11 +11,7 @@ runtime.python_library(
1111 "//executorch/...",
1212 ],
1313 deps = [
14- "//caffe2:torch",
15- "//executorch/backends/aoti/passes:passes",
16- "//executorch/exir/_serialize:lib",
17- "//executorch/exir/backend:backend_details",
18- "//executorch/exir/backend:compile_spec_schema",
14+ "//executorch/backends/aoti:aoti_backend",
1915 ],
2016)
2117
@@ -29,7 +25,6 @@ runtime.python_library(
2925 ],
3026 deps = [
3127 "//caffe2:torch",
32- "//executorch/exir/backend:partitioner",
33- "//executorch/exir/backend:utils",
28+ "//executorch/backends/aoti:aoti_partitioner",
3429 ],
3530)
You can’t perform that action at this time.
0 commit comments