xref: /aosp_15_r20/external/executorch/examples/models/llama/experimental/targets.bzl (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
2
3def define_common_targets():
4    runtime.python_library(
5        name = "subclass",
6        srcs = [
7            "subclass.py",
8        ],
9        deps = [
10            "//caffe2:torch",
11            "//pytorch/ao:torchao",
12        ],
13    )
14
15    runtime.python_test(
16        name = "test_subclass",
17        srcs = [
18            "test_subclass.py",
19        ],
20        deps = [
21            ":subclass",
22        ],
23    )
24