Skip to content

Commit dfefe60

Browse files
committed
Update on "[aoti-backend-consolidation 1/3] partitioners"
Differential Revision: [D85700449](https://our.internmc.facebook.com/intern/diff/D85700449/) [ghstack-poisoned]
1 parent b33c284 commit dfefe60

File tree

2 files changed

+16
-2
lines changed

2 files changed

+16
-2
lines changed

backends/aoti/targets.bzl

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,21 @@
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+
419
# AOTI common shims functionality
520
runtime.cxx_library(
621
name = "common_shims",

backends/cuda/TARGETS

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@ runtime.python_library(
2929
],
3030
deps = [
3131
"//caffe2:torch",
32-
"//executorch/exir/backend:partitioner",
33-
"//executorch/exir/backend:utils",
32+
"//executorch/backends/aoti:aoti_partitioner",
3433
],
3534
)

0 commit comments

Comments
 (0)