xref: /aosp_15_r20/external/executorch/test/end2end/TARGETS (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a) 
1# @noautodeps
2
3load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
4load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest")
5
6oncall("executorch")
7
8python_library(
9    name = "exported_module",
10    srcs = [
11        "exported_module.py",
12    ],
13    visibility = [
14        "//executorch/...",
15    ],
16    deps = [
17        "//caffe2:torch",
18        "//executorch/exir:dynamic_shape",
19        "//executorch/exir:lib",
20        "//executorch/exir:pass_manager",
21        "//executorch/exir/passes:lib",
22    ],
23)
24
25python_library(
26    name = "register_scratch_meta_fns",
27    srcs = [
28        "register_scratch_meta_fns.py",
29    ],
30    deps = [
31        "//executorch/exir/operator:manip",
32    ],
33)
34
35python_unittest(
36    name = "end2end_aten",
37    srcs = [
38        "test_end2end.py",
39    ],
40    preload_deps = ["//executorch/kernels/portable:custom_ops_generated_lib"],
41    deps = [
42        ":exported_module",
43        ":register_scratch_meta_fns",
44        "//caffe2:torch",
45        "//executorch/devtools:lib",
46        "//executorch/devtools/bundled_program:config",
47        "//executorch/devtools/bundled_program/serialize:lib",
48        "//executorch/exir:dynamic_shape",
49        "//executorch/exir:lib",
50        "//executorch/exir:memory",
51        "//executorch/exir:pass_manager",
52        "//executorch/exir:print_program",
53        "//executorch/exir:tensor",
54        "//executorch/exir/_serialize:lib",
55        "//executorch/exir/emit:lib",
56        "//executorch/exir/passes:lib",
57        "//executorch/exir/tests:control_flow_models",
58        "//executorch/exir/tests:dynamic_shape_models",
59        "//executorch/exir/tests:models",
60        "//executorch/exir/tests:transformer",
61        "//executorch/extension/pybindings:aten_lib",
62        "//executorch/extension/pytree:pybindings",
63    ],
64)
65
66python_unittest(
67    name = "end2end",
68    srcs = [
69        "test_end2end.py",
70    ],
71    preload_deps = ["//executorch/kernels/portable:custom_ops_generated_lib"],
72    deps = [
73        ":exported_module",
74        ":register_scratch_meta_fns",
75        "//caffe2:torch",
76        "//executorch/devtools:lib",
77        "//executorch/devtools/bundled_program:config",
78        "//executorch/devtools/bundled_program/serialize:lib",
79        "//executorch/exir:dynamic_shape",
80        "//executorch/exir:lib",
81        "//executorch/exir:memory",
82        "//executorch/exir:pass_manager",
83        "//executorch/exir:print_program",
84        "//executorch/exir:tensor",
85        "//executorch/exir/_serialize:lib",
86        "//executorch/exir/emit:lib",
87        "//executorch/exir/passes:lib",
88        "//executorch/exir/tests:control_flow_models",
89        "//executorch/exir/tests:dynamic_shape_models",
90        "//executorch/exir/tests:models",
91        "//executorch/exir/tests:transformer",
92        "//executorch/extension/pybindings:portable_lib",
93        "//executorch/extension/pytree:pybindings",
94    ],
95)
96