xref: /aosp_15_r20/external/executorch/extension/llm/export/TARGETS (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1# Any targets that should be shared between fbcode and xplat must be defined in
2# targets.bzl. This file can contain fbcode-only targets.
3
4load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
5load(":targets.bzl", "define_common_targets")
6
7oncall("executorch")
8
9define_common_targets()
10
11runtime.python_library(
12    name = "export_lib",
13    srcs = [
14        "builder.py",
15        "partitioner_lib.py",
16        "quantizer_lib.py",
17    ],
18    _is_external_target = True,
19    base_module = "executorch.extension.llm.export",
20    visibility = [
21        "//bento/...",
22        "//bento_kernels/...",
23        "//executorch/examples/...",
24    ],
25    deps = [
26        "//caffe2:torch",
27        "//executorch/backends/apple/coreml:backend",
28        "//executorch/backends/apple/coreml:partitioner",
29        "//executorch/backends/apple/mps:partitioner",
30        "//executorch/backends/qualcomm/partition:partition",
31        "//executorch/backends/qualcomm/quantizer:quantizer",
32        "//executorch/backends/transforms:duplicate_dynamic_quant_chain",
33        "//executorch/backends/vulkan/partitioner:vulkan_partitioner",
34        "//executorch/backends/vulkan/quantizer:vulkan_quantizer",
35        "//executorch/backends/xnnpack/partition:xnnpack_partitioner",
36        "//executorch/exir:lib",
37        "//executorch/exir/backend:backend_details",
38        "//executorch/extension/export_util:export_util",
39        "//executorch/extension/llm/tokenizer:tokenizer_py_lib",
40    ],
41)
42