File tree Expand file tree Collapse file tree 3 files changed +31
-1
lines changed Expand file tree Collapse file tree 3 files changed +31
-1
lines changed Original file line number Diff line number Diff line change @@ -54,3 +54,20 @@ runtime.python_library(
54
54
"//executorch/exir/backend:partitioner",
55
55
],
56
56
)
57
+
58
+ runtime.python_library(
59
+ name = "group_partitioner_lib",
60
+ srcs = [
61
+ "group_partitioner.py",
62
+ ],
63
+ visibility = [
64
+ "//executorch/...",
65
+ "//executorch/exir/backend/...",
66
+ "//executorch/test/...",
67
+ "@EXECUTORCH_CLIENTS",
68
+ ],
69
+ deps = [
70
+ "//caffe2:torch",
71
+ "//executorch/exir/backend:partitioner",
72
+ ],
73
+ )
Original file line number Diff line number Diff line change @@ -447,3 +447,14 @@ python_unittest(
447
447
"//executorch/extension/pybindings:portable_lib", # @manual
448
448
],
449
449
)
450
+
451
+ python_unittest(
452
+ name = "test_group_partitioner",
453
+ srcs = [
454
+ "test_group_partitioner.py",
455
+ ],
456
+ deps = [
457
+ "//caffe2:torch",
458
+ "//executorch/exir/backend/canonical_partitioners:group_partitioner_lib",
459
+ ],
460
+ )
Original file line number Diff line number Diff line change 8
8
from typing import List
9
9
10
10
import torch
11
- from exir .backend .canonical_partitioners .group_partitioner import GroupBasedPartitioner
11
+ from executorch .exir .backend .canonical_partitioners .group_partitioner import (
12
+ GroupBasedPartitioner ,
13
+ )
12
14
13
15
from torch .fx .passes .infra .partitioner import CapabilityBasedPartitioner
14
16
from torch .fx .passes .operator_support import OperatorSupportBase
You can’t perform that action at this time.
0 commit comments