xref: /aosp_15_r20/external/executorch/exir/backend/test/demos/TARGETS (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a) 
1load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest")
2
3python_unittest(
4    name = "test_xnnpack_qnnpack",
5    srcs = [
6        "test_xnnpack_qnnpack.py",
7    ],
8    preload_deps = [
9        "//executorch/kernels/portable:custom_ops_generated_lib",
10        "//executorch/kernels/quantized:custom_ops_generated_lib",
11    ],
12    deps = [
13        "//caffe2:torch",
14        "//executorch/backends/fb/qnnpack:qnnpack_backend",
15        "//executorch/backends/fb/qnnpack:qnnpack_preprocess",
16        "//executorch/backends/fb/qnnpack/partition:qnnpack_partitioner",
17        "//executorch/backends/xnnpack:xnnpack_backend",
18        "//executorch/backends/xnnpack:xnnpack_preprocess",
19        "//executorch/backends/xnnpack/partition:xnnpack_partitioner",
20        "//executorch/exir:lib",
21        "//executorch/exir/backend:backend_api",
22        "//executorch/exir/passes:spec_prop_pass",
23        "//executorch/extension/pybindings:portable_lib",  # @manual
24        "//executorch/extension/pytree:pylib",
25    ],
26)
27
28python_unittest(
29    name = "test_delegate_aten_mode",
30    srcs = [
31        "test_delegate_aten_mode.py",
32    ],
33    preload_deps = [
34        "//executorch/kernels/portable:custom_ops_generated_lib_aten",
35    ],
36    deps = [
37        "//caffe2:torch",
38        "//executorch/exir:lib",
39        "//executorch/exir/backend:backend_api",
40        "//executorch/exir/backend:compile_spec_schema",
41        "//executorch/exir/backend/test:backend_with_compiler_demo",
42        "//executorch/extension/pybindings:aten_lib",  # @manual
43        "//executorch/extension/pytree:pylib",
44    ],
45)
46