xref: /aosp_15_r20/external/executorch/examples/models/llama/targets.bzl (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a) 
1load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "get_oss_build_kwargs", "runtime")
2
3def define_common_targets():
4    for aten in (True, False):
5        if not runtime.is_oss or not aten:
6            aten_suffix = "_aten" if aten else ""
7
8            runtime.cxx_binary(
9                name = "main" + aten_suffix,
10                srcs = [
11                    "main.cpp",
12                ],
13                compiler_flags = ["-Wno-global-constructors"],
14                preprocessor_flags = [
15                    "-DUSE_ATEN_LIB",
16                ] if aten else [],
17                deps = [
18                    "//executorch/examples/models/llama/runner:runner" + aten_suffix,
19                    "//executorch/extension/evalue_util:print_evalue",
20                    "//executorch/extension/threadpool:threadpool",
21                    "//executorch/extension/threadpool:cpuinfo_utils",
22                ],
23                external_deps = [
24                    "gflags",
25                ],
26                **get_oss_build_kwargs()
27            )
28