File tree Expand file tree Collapse file tree 2 files changed +71
-1
lines changed
Expand file tree Collapse file tree 2 files changed +71
-1
lines changed Original file line number Diff line number Diff line change 1+ load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
2+
3+ oncall("executorch")
4+
5+ python_library(
6+ name = "neutron_edge_pass",
7+ srcs = [
8+ "neutron_edge_pass.py",
9+ ],
10+ deps = [
11+ "//executorch/exir:lib",
12+ "//caffe2:torch",
13+ ],
14+ )
15+
16+ python_library(
17+ name = "move_auxiliary_operator_into_separate_qdq_cluster_pass",
18+ srcs = [
19+ "move_auxiliary_operator_into_separate_qdq_cluster_pass.py",
20+ ],
21+ deps = [
22+ ":neutron_edge_pass",
23+ "//executorch/backends/nxp:neutron_backend",
24+ "//executorch/exir:lib",
25+ "//caffe2:torch",
26+ ],
27+ )
28+
29+ python_library(
30+ name = "neutron_edge_pass_manager",
31+ srcs = [
32+ "neutron_edge_pass_manager.py",
33+ ],
34+ deps = [
35+ ":neutron_edge_pass",
36+ ":move_auxiliary_operator_into_separate_qdq_cluster_pass",
37+ "//executorch/exir:lib",
38+ "//caffe2:torch",
39+ ],
40+ )
Original file line number Diff line number Diff line change @@ -21,9 +21,11 @@ python_library(
2121 ],
2222 deps = [
2323 "//executorch/exir:lib",
24- "//pytorch/ao:torchao",
24+ "//pytorch/ao:torchao",
2525 "//executorch/backends/nxp:quantizer",
2626 "//executorch/backends/nxp:neutron_backend",
27+ "//executorch/backends/nxp/edge_passes:neutron_edge_pass_manager",
28+ "//executorch/extension/export_util:export_util",
2729 ]
2830)
2931
@@ -50,3 +52,31 @@ python_pytest(
5052 ":models",
5153 ]
5254)
55+
56+ python_pytest(
57+ name = "test_edge_passes",
58+ srcs = [
59+ "test_edge_passes.py",
60+ ],
61+ deps = [
62+ "//executorch/backends/nxp:neutron_backend",
63+ ":executorch_pipeline",
64+ ":models",
65+ "//executorch/exir:lib",
66+ "//caffe2:torch",
67+ ]
68+ )
69+
70+ python_pytest(
71+ name = "test_batch_norm_fusion",
72+ srcs = [
73+ "test_batch_norm_fusion.py",
74+ ],
75+ deps = [
76+ "//executorch/backends/nxp:neutron_backend",
77+ ":executorch_pipeline",
78+ ":models",
79+ "//executorch/exir:lib",
80+ "//caffe2:torch",
81+ ]
82+ )
You can’t perform that action at this time.
0 commit comments