xref: /aosp_15_r20/external/executorch/extension/llm/sampler/targets.bzl (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
2
3def define_common_targets():
4    for aten in (True, False):
5        aten_suffix = "_aten" if aten else ""
6
7        runtime.cxx_library(
8            name = "sampler" + aten_suffix,
9            exported_headers = [
10                "sampler.h",
11            ],
12            preprocessor_flags = [
13                "-DUSE_ATEN_LIB",
14            ] if aten else [],
15            srcs = [
16                "sampler.cpp",
17            ],
18            visibility = [
19                "@EXECUTORCH_CLIENTS",
20            ],
21            external_deps = [
22                "libtorch",
23            ] if aten else [],
24            exported_deps = [
25                "//executorch/runtime/core/exec_aten:lib" + aten_suffix,
26                "//executorch/runtime/platform:compiler",
27            ],
28        )
29