-
Notifications
You must be signed in to change notification settings - Fork 528
/
Copy pathTARGETS
108 lines (108 loc) · 2.85 KB
/
TARGETS
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
# @noautodeps
load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
python_library(
name = "arm_partitioner",
srcs = [
"ethosu_backend.py",
"ethosu_partitioner.py",
"tosa_backend.py",
"tosa_partitioner.py",
],
deps = [
":arm_backend",
"//executorch/backends/arm/operator_support:operator_support",
"//executorch/backends/arm/_passes:passes",
"//executorch/exir:lib",
],
)
python_library(
name = "arm_backend",
srcs = [
"arm_backend.py",
],
deps = [
"fbsource//third-party/pypi/flatbuffers:flatbuffers",
"fbsource//third-party/pypi/ml-dtypes:ml-dtypes",
"fbsource//third-party/serialization_lib/python/serializer:serializer",
"fbsource//third-party/serialization_lib/python/tosa:tosa",
":arm_vela",
":process_node",
"//executorch/backends/arm/operators:lib",
"//executorch/backends/arm/operators:node_visitor",
"//executorch/backends/arm/_passes:passes",
],
)
python_library(
name = "process_node",
srcs = ["process_node.py"],
deps = [
"fbsource//third-party/serialization_lib/python/tosa:tosa",
"//executorch/backends/arm/operators:node_visitor",
"//executorch/backends/arm:tosa_mapping",
"//executorch/backends/arm:tosa_quant_utils",
"//executorch/backends/arm:tosa_utils",
"//executorch/exir:lib",
],
)
python_library(
name = "arm_vela",
srcs = [
"arm_vela.py",
],
deps = [
"fbsource//third-party/pypi/ethos-u-vela:ethos-u-vela",
],
)
python_library(
name = "tosa_mapping",
srcs = [
"tosa_mapping.py",
],
deps = [
"fbsource//third-party/serialization_lib/python/serializer:serializer",
"//caffe2:torch",
],
)
python_library(
name = "tosa_quant_utils",
srcs = [
"tosa_quant_utils.py",
],
deps = [
"fbsource//third-party/pypi/numpy:numpy",
"fbsource//third-party/serialization_lib/python/serializer:serializer",
"fbsource//third-party/serialization_lib/python/tosa:tosa",
":tosa_mapping",
"//executorch/exir/dialects:lib",
],
)
python_library(
name = "tosa_specification",
srcs = [
"tosa_specification.py",
],
deps = [
"fbsource//third-party/pypi/packaging:packaging",
"//executorch/exir/backend:compile_spec_schema",
],
)
python_library(
name = "tosa_utils",
srcs = [
"tosa_utils.py",
],
deps = [
"fbsource//third-party/serialization_lib/python/serializer:serializer",
":tosa_quant_utils",
"//executorch/backends/arm/operators:node_visitor",
],
)
python_library(
name = "arm_model_evaluator",
srcs = [
"util/arm_model_evaluator.py",
],
deps = [
"//caffe2:torch",
]
)