executorch/backends/qnnpack/test/TARGETS at b580e71b20e71c656ae70798ac6a54967747a75f · pytorch/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
load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest")
load(":targets.bzl", "define_common_targets")

define_common_targets()

python_unittest(
    name = "test_qnnpack",
    srcs = [
        "test_qnnpack.py",
    ],
    preload_deps = [
        "//executorch/kernels/quantized:custom_ops_generated_lib",
    ],
    supports_static_listing = True,
    tags = ["long_running"],
    deps = [
        "//caffe2:torch",
        "//executorch/backends:backend_api",
        "//executorch/backends/qnnpack:qnnpack_backend",
        "//executorch/backends/qnnpack:qnnpack_preprocess",
        "//executorch/backends/qnnpack/partition:qnnpack_partitioner",
        "//executorch/exir:lib",
        "//executorch/exir/serialize:lib",
        "//executorch/pybindings:portable",  # @manual
        "//executorch/pytree:pylib",
    ],
)

python_unittest(
    name = "test_qnnpack_partitioner",
    srcs = [
        "test_qnnpack_partitioner.py",
    ],
    preload_deps = [
        "//executorch/kernels/quantized:custom_ops_generated_lib",
    ],
    supports_static_listing = True,
    tags = ["long_running"],
    deps = [
        "//caffe2:torch",
        "//executorch/backends/qnnpack:qnnpack_backend",
        "//executorch/backends/qnnpack/partition:support_patterns",
        "//executorch/backends/transforms:lib",
        "//executorch/exir:lib",
        "//executorch/pybindings:portable",  # @manual
    ],
)