executorch/extension/android/BUCK at main · CodeLinaro/executorch · 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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
load("@fbcode_macros//build_defs:build_file_migration.bzl", "fbcode_target", "non_fbcode_target")
load("@fbsource//tools/build_defs/android:fb_android_library.bzl", "fb_android_library")

oncall("executorch")

non_fbcode_target(_kind = fb_android_library,
    name = "executorch",
    warnings_as_errors = False,
    required_for_source_only_abi = True,
    srcs = [
        "executorch_android/src/main/java/org/pytorch/executorch/DType.java",
        "executorch_android/src/main/java/org/pytorch/executorch/EValue.java",
        "executorch_android/src/main/java/org/pytorch/executorch/ExecuTorchRuntime.java",
        "executorch_android/src/main/java/org/pytorch/executorch/ExecutorchRuntimeException.java",
        "executorch_android/src/main/java/org/pytorch/executorch/MethodMetadata.java",
        "executorch_android/src/main/java/org/pytorch/executorch/Module.java",
        "executorch_android/src/main/java/org/pytorch/executorch/Tensor.java",
        "executorch_android/src/main/java/org/pytorch/executorch/annotations/Experimental.java",
    ],
    autoglob = False,
    language = "JAVA",
    deps = [
        "//fbandroid/java/com/facebook/jni:jni",
        "//fbandroid/libraries/soloader/java/com/facebook/soloader/nativeloader:nativeloader",
    ] + ([
        "//xplat/executorch/backends/vulkan:vulkan_backend_lib_static",
    ] if read_config("executorch", "minimal_jni", "false") == "false" else []),
)

non_fbcode_target(_kind = fb_android_library,
    name = "executorch_training",
    warnings_as_errors = False,
    srcs = [
        "executorch_android/src/main/java/org/pytorch/executorch/training/SGD.java",
        "executorch_android/src/main/java/org/pytorch/executorch/training/TrainingModule.java",
    ],
    autoglob = False,
    language = "JAVA",
    deps = [
        ":executorch",
        "//fbandroid/java/com/facebook/jni:jni",
        "//fbandroid/libraries/soloader/java/com/facebook/soloader/nativeloader:nativeloader",
    ],
)

non_fbcode_target(_kind = fb_android_library,
    name = "executorch_llama",
    warnings_as_errors = False,
    srcs = [
        "executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmCallback.java",
        "executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmGenerationConfig.java",
        "executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmModule.java",
        "executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmModuleConfig.java",
    ],
    autoglob = False,
    language = "JAVA",
    deps = [
        ":executorch",
        "//fbandroid/java/com/facebook/jni:jni",
        "//fbandroid/libraries/soloader/java/com/facebook/soloader/nativeloader:nativeloader",
    ],
)