Skip to content

Commit 65164f7

Browse files
manuelcandalesfacebook-github-bot
authored andcommitted
NXP Backend: Forward fix pytorch#13183
Differential Revision: D80279279
1 parent b1d09fd commit 65164f7

File tree

2 files changed

+71
-1
lines changed

2 files changed

+71
-1
lines changed

backends/nxp/edge_passes/TARGETS

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
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+
)

backends/nxp/tests/TARGETS

Lines changed: 31 additions & 1 deletion
Original file line numberDiff line numberDiff 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+
)

0 commit comments

Comments
 (0)