xref: /aosp_15_r20/external/executorch/examples/models/llama/runner/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 = "eager_runner_library",
13    srcs = [
14        "eager.py",
15        "generation.py"
16    ],
17    _is_external_target = True,
18    base_module = "executorch.examples.models.llama.runner",
19    visibility = [
20        "//bento/...",
21        "//bento_kernels/...",
22        "//executorch/examples/...",
23        "@EXECUTORCH_CLIENTS",
24    ],
25    deps = [
26        "//executorch/examples/models/llama:export_library",
27    ],
28)
29
30runtime.python_binary(
31    name = "eager",
32    main_function = "executorch.examples.models.llama.runner.eager.main",
33    deps = [
34        ":eager_runner_library",
35        "//caffe2:torch",
36    ],
37)
38