xref: /aosp_15_r20/external/executorch/extension/pytree/TARGETS (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1*523fa7a6SAndroid Build Coastguard Worker# Any targets that should be shared between fbcode and xplat must be defined in
2*523fa7a6SAndroid Build Coastguard Worker# targets.bzl. This file can contain fbcode-only targets.
3*523fa7a6SAndroid Build Coastguard Worker
4*523fa7a6SAndroid Build Coastguard Workerload("@fbcode_macros//build_defs:cpp_python_extension.bzl", "cpp_python_extension")
5*523fa7a6SAndroid Build Coastguard Workerload("@fbcode_macros//build_defs:python_library.bzl", "python_library")
6*523fa7a6SAndroid Build Coastguard Workerload(":targets.bzl", "define_common_targets")
7*523fa7a6SAndroid Build Coastguard Worker
8*523fa7a6SAndroid Build Coastguard Workeroncall("executorch")
9*523fa7a6SAndroid Build Coastguard Worker
10*523fa7a6SAndroid Build Coastguard Workerdefine_common_targets()
11*523fa7a6SAndroid Build Coastguard Worker
12*523fa7a6SAndroid Build Coastguard Workercpp_python_extension(
13*523fa7a6SAndroid Build Coastguard Worker    name = "pybindings",
14*523fa7a6SAndroid Build Coastguard Worker    srcs = [
15*523fa7a6SAndroid Build Coastguard Worker        "pybindings.cpp",
16*523fa7a6SAndroid Build Coastguard Worker    ],
17*523fa7a6SAndroid Build Coastguard Worker    base_module = "executorch.extension.pytree",
18*523fa7a6SAndroid Build Coastguard Worker    deps = [
19*523fa7a6SAndroid Build Coastguard Worker        ":pytree",
20*523fa7a6SAndroid Build Coastguard Worker    ],
21*523fa7a6SAndroid Build Coastguard Worker    external_deps = [
22*523fa7a6SAndroid Build Coastguard Worker        "pybind11",
23*523fa7a6SAndroid Build Coastguard Worker    ],
24*523fa7a6SAndroid Build Coastguard Worker)
25*523fa7a6SAndroid Build Coastguard Worker
26*523fa7a6SAndroid Build Coastguard Workercpp_python_extension(
27*523fa7a6SAndroid Build Coastguard Worker    name = "pybindings_debug",
28*523fa7a6SAndroid Build Coastguard Worker    srcs = [
29*523fa7a6SAndroid Build Coastguard Worker        "pybindings.cpp",
30*523fa7a6SAndroid Build Coastguard Worker    ],
31*523fa7a6SAndroid Build Coastguard Worker    base_module = "executorch.extension.pytree",
32*523fa7a6SAndroid Build Coastguard Worker    deps = [
33*523fa7a6SAndroid Build Coastguard Worker        ":pytree",
34*523fa7a6SAndroid Build Coastguard Worker    ],
35*523fa7a6SAndroid Build Coastguard Worker    external_deps = [
36*523fa7a6SAndroid Build Coastguard Worker        "pybind11",
37*523fa7a6SAndroid Build Coastguard Worker    ],
38*523fa7a6SAndroid Build Coastguard Worker)
39*523fa7a6SAndroid Build Coastguard Worker
40*523fa7a6SAndroid Build Coastguard Workerpython_library(
41*523fa7a6SAndroid Build Coastguard Worker    name = "pylib",
42*523fa7a6SAndroid Build Coastguard Worker    srcs = [
43*523fa7a6SAndroid Build Coastguard Worker        "__init__.py",
44*523fa7a6SAndroid Build Coastguard Worker    ],
45*523fa7a6SAndroid Build Coastguard Worker    base_module = "executorch.extension.pytree",
46*523fa7a6SAndroid Build Coastguard Worker    deps = [
47*523fa7a6SAndroid Build Coastguard Worker        ":pybindings",
48*523fa7a6SAndroid Build Coastguard Worker        "//caffe2:torch",
49*523fa7a6SAndroid Build Coastguard Worker    ],
50*523fa7a6SAndroid Build Coastguard Worker)
51