xref: /aosp_15_r20/external/executorch/extension/module/test/targets.bzl (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1*523fa7a6SAndroid Build Coastguard Workerload(
2*523fa7a6SAndroid Build Coastguard Worker    "@fbsource//tools/build_defs:default_platform_defs.bzl",
3*523fa7a6SAndroid Build Coastguard Worker    "ANDROID",
4*523fa7a6SAndroid Build Coastguard Worker    "CXX",
5*523fa7a6SAndroid Build Coastguard Worker)
6*523fa7a6SAndroid Build Coastguard Workerload("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
7*523fa7a6SAndroid Build Coastguard Worker
8*523fa7a6SAndroid Build Coastguard Workerdef define_common_targets():
9*523fa7a6SAndroid Build Coastguard Worker    """Defines targets that should be shared between fbcode and xplat.
10*523fa7a6SAndroid Build Coastguard Worker
11*523fa7a6SAndroid Build Coastguard Worker    The directory containing this targets.bzl file should also contain both
12*523fa7a6SAndroid Build Coastguard Worker    TARGETS and BUCK files that call this function.
13*523fa7a6SAndroid Build Coastguard Worker    """
14*523fa7a6SAndroid Build Coastguard Worker
15*523fa7a6SAndroid Build Coastguard Worker    for aten_mode in (True, False):
16*523fa7a6SAndroid Build Coastguard Worker        aten_suffix = ("_aten" if aten_mode else "")
17*523fa7a6SAndroid Build Coastguard Worker
18*523fa7a6SAndroid Build Coastguard Worker        runtime.cxx_test(
19*523fa7a6SAndroid Build Coastguard Worker            name = "test" + aten_suffix,
20*523fa7a6SAndroid Build Coastguard Worker            srcs = [
21*523fa7a6SAndroid Build Coastguard Worker                "module_test.cpp",
22*523fa7a6SAndroid Build Coastguard Worker            ],
23*523fa7a6SAndroid Build Coastguard Worker            deps = [
24*523fa7a6SAndroid Build Coastguard Worker                "//executorch/kernels/portable:generated_lib" + aten_suffix,
25*523fa7a6SAndroid Build Coastguard Worker                "//executorch/extension/data_loader:file_data_loader",
26*523fa7a6SAndroid Build Coastguard Worker                "//executorch/extension/module:module" + aten_suffix,
27*523fa7a6SAndroid Build Coastguard Worker                "//executorch/extension/tensor:tensor" + aten_suffix,
28*523fa7a6SAndroid Build Coastguard Worker            ],
29*523fa7a6SAndroid Build Coastguard Worker            env = {
30*523fa7a6SAndroid Build Coastguard Worker                "RESOURCES_PATH": "$(location :resources)/resources",
31*523fa7a6SAndroid Build Coastguard Worker            },
32*523fa7a6SAndroid Build Coastguard Worker            platforms = [CXX, ANDROID],  # Cannot bundle resources on Apple platform.
33*523fa7a6SAndroid Build Coastguard Worker            compiler_flags = [
34*523fa7a6SAndroid Build Coastguard Worker                "-Wno-error=deprecated-declarations",
35*523fa7a6SAndroid Build Coastguard Worker            ],
36*523fa7a6SAndroid Build Coastguard Worker        )
37*523fa7a6SAndroid Build Coastguard Worker
38*523fa7a6SAndroid Build Coastguard Worker    runtime.filegroup(
39*523fa7a6SAndroid Build Coastguard Worker        name = "resources",
40*523fa7a6SAndroid Build Coastguard Worker        srcs = native.glob([
41*523fa7a6SAndroid Build Coastguard Worker            "resources/**",
42*523fa7a6SAndroid Build Coastguard Worker        ]),
43*523fa7a6SAndroid Build Coastguard Worker    )
44