executorch-1/examples/models/llama2/tests/TARGETS at e4a6ccbdcca5f42c36b2ddf4e57bc15141841c82 · mergennachin/executorch-1 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest")

oncall("executorch")

python_unittest(
    name = "test_simple_sdpa",
    srcs = [
        "test_simple_sdpa.py",
    ],
    deps = [
        "//caffe2:torch",
        "//executorch/examples/models/llama2:export_library",
        "//executorch/examples/models/llama2:llama_transformer",
    ],
)

python_unittest(
    name = "test_spinquant_transforms",
    srcs = [
        "test_spinquant_transforms.py",
    ],
    deps = [
        "//caffe2:torch",
        "//executorch/examples/models/llama2:export_library",
        "//executorch/examples/models/llama2:llama_transformer",
        "//pytorch/ao:torchao",
    ],
)