Skip to content

Commit 0d9353a

Browse files
committed
Update on "[aoti-backend-consolidation 2/3] backend.py"
Differential Revision: [D85704977](https://our.internmc.facebook.com/intern/diff/D85704977/) [ghstack-poisoned]
2 parents 06abff6 + 1240eae commit 0d9353a

File tree

2 files changed

+34
-7
lines changed

2 files changed

+34
-7
lines changed

backends/aoti/targets.bzl

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,38 @@
11
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
22

33
def 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",

backends/cuda/TARGETS

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff 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
)

0 commit comments

Comments
 (0)