xref: /aosp_15_r20/external/executorch/extension/android/BUCK (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1load("@fbsource//tools/build_defs/android:fb_android_library.bzl", "fb_android_library")
2
3oncall("executorch")
4
5fb_android_library(
6    name = "executorch",
7    srcs = [
8        "src/main/java/org/pytorch/executorch/DType.java",
9        "src/main/java/org/pytorch/executorch/EValue.java",
10        "src/main/java/org/pytorch/executorch/Module.java",
11        "src/main/java/org/pytorch/executorch/NativePeer.java",
12        "src/main/java/org/pytorch/executorch/Tensor.java",
13        "src/main/java/org/pytorch/executorch/annotations/Experimental.java",
14    ],
15    autoglob = False,
16    language = "JAVA",
17    deps = [
18        "//fbandroid/java/com/facebook/jni:jni",
19        "//fbandroid/libraries/soloader/java/com/facebook/soloader/nativeloader:nativeloader",
20    ],
21)
22
23fb_android_library(
24    name = "executorch_llama",
25    srcs = [
26        "src/main/java/org/pytorch/executorch/DType.java",
27        "src/main/java/org/pytorch/executorch/EValue.java",
28        "src/main/java/org/pytorch/executorch/LlamaCallback.java",
29        "src/main/java/org/pytorch/executorch/LlamaModule.java",
30        "src/main/java/org/pytorch/executorch/Module.java",
31        "src/main/java/org/pytorch/executorch/NativePeer.java",
32        "src/main/java/org/pytorch/executorch/Tensor.java",
33        "src/main/java/org/pytorch/executorch/annotations/Experimental.java",
34    ],
35    autoglob = False,
36    language = "JAVA",
37    deps = [
38        "//fbandroid/java/com/facebook/jni:jni",
39        "//fbandroid/libraries/soloader/java/com/facebook/soloader/nativeloader:nativeloader",
40    ],
41)
42