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