xref: /aosp_15_r20/external/executorch/examples/models/llama/tests/TARGETS (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a) 
1load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest")
2
3oncall("executorch")
4
5python_unittest(
6    name = "test_simple_sdpa",
7    srcs = [
8        "test_simple_sdpa.py",
9    ],
10    deps = [
11        "//caffe2:torch",
12        "//executorch/examples/models/llama:export_library",
13        "//executorch/examples/models/llama:llama_transformer",
14    ],
15)
16
17python_unittest(
18    name = "test_pre_quantization_transforms",
19    srcs = [
20        "test_pre_quantization_transforms.py",
21    ],
22    deps = [
23        "//caffe2:torch",
24        "//executorch/examples/models/llama:export_library",
25        "//executorch/examples/models/llama:llama_transformer",
26        "//pytorch/ao:torchao",
27    ],
28)
29