Skip to content

Commit 160d55c

Browse files
Arm backend: Add experimental support for new TOSAQuantizer
Differential Revision: D96368239 Pull Request resolved: #18100
1 parent 94e9ca6 commit 160d55c

16 files changed

+1775
-889
lines changed

backends/arm/quantizer/TARGETS

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,16 +10,17 @@ runtime.python_library(
1010
],
1111
)
1212

13-
# Exposed through __init__.py
1413
runtime.python_library(
1514
name = "arm_quantizer",
1615
srcs = ["arm_quantizer.py"],
1716
deps = [
1817
":arm_quantizer_utils",
1918
":quantization_annotator",
19+
":quantizer_support",
2020
"//executorch/backends/arm:constants",
2121
"//executorch/backends/arm:ethosu",
2222
"//executorch/backends/arm:vgf",
23+
"//executorch/backends/cortex_m/quantizer:quantizer",
2324
"//executorch/backends/arm/tosa:specification",
2425
"//executorch/backends/arm:arm_compile_spec",
2526
"//caffe2:torch",
@@ -43,11 +44,24 @@ runtime.python_library(
4344
name = "arm_quantizer_utils",
4445
srcs = ["arm_quantizer_utils.py"],
4546
deps = [
47+
"//caffe2:torch",
48+
"//executorch/backends/arm:common",
49+
"//executorch/backends/arm:constants",
4650
":quantization_config",
4751
"//pytorch/ao:torchao",
4852
],
4953
)
5054

55+
runtime.python_library(
56+
name = "quantizer_support",
57+
srcs = ["quantizer_support.py"],
58+
deps = [
59+
":quantization_annotator",
60+
"//caffe2:torch",
61+
"//executorch/backends/cortex_m/quantizer:quantizer",
62+
],
63+
)
64+
5165
runtime.python_library(
5266
name = "lib",
5367
srcs = ["__init__.py"],

0 commit comments

Comments
 (0)