xref: /aosp_15_r20/external/executorch/extension/llm/custom_ops/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_test(
12    name = "test_sdpa_with_kv_cache",
13    srcs = [
14        "test_sdpa_with_kv_cache.py",
15    ],
16    preload_deps = [
17        ":custom_ops_aot_lib_mkl_noomp",
18        ":custom_ops_aot_py",
19    ],
20    deps = [
21        "//caffe2:torch",
22    ],
23)
24
25runtime.python_test(
26    name = "test_update_quantized_cache",
27    srcs = [
28        "test_update_quantized_cache.py",
29    ],
30    preload_deps = [
31        ":custom_ops_aot_lib",
32    ],
33    deps = [
34        "//caffe2:torch",
35    ],
36)
37
38runtime.python_test(
39    name = "test_preprocess_custom_ops",
40    srcs = [
41        "test_preprocess_custom_ops.py",
42    ],
43    preload_deps = [
44        ":preprocess_custom_ops_py",
45    ],
46    deps = [
47        "//caffe2:torch",
48    ],
49)
50