xref: /aosp_15_r20/external/executorch/kernels/quantized/test/TARGETS (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a) 
1load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest")
2load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
3load(":targets.bzl", "define_common_targets")
4
5oncall("executorch")
6
7define_common_targets()
8
9python_unittest(
10    name = "test_out_variants",
11    srcs = ["test_out_variants.py"],
12    preload_deps = [
13        "//executorch/kernels/quantized:custom_ops_generated_lib",
14    ],
15    deps = [
16        "//caffe2:torch",
17        "//executorch/exir/dialects:lib",
18        "//executorch/exir/passes:quant_fusion_pass",
19        "//executorch/kernels/quantized:quantized_ops_lib",
20    ],
21)
22
23runtime.cxx_library(
24    name = "quantized_ops_for_test_lib",
25    srcs = [
26        "quantized_ops_aot_register.cpp",
27    ],
28    visibility = [
29        "//executorch/...",
30        "@EXECUTORCH_CLIENTS",
31    ],
32    deps = [
33        "//executorch/extension/aten_util:aten_bridge",
34        "//executorch/kernels/quantized/cpu:op_dequantize",
35        "//executorch/kernels/quantized/cpu:op_quantize",
36        "//executorch/runtime/core/exec_aten:lib",
37    ],
38    external_deps = [
39        "libtorch",
40    ],
41)
42
43python_unittest(
44    name = "test_quant_dequant_per_token",
45    srcs = [
46        "test_quant_dequant_per_token.py",
47    ],
48    preload_deps = [
49        ":quantized_ops_for_test_lib",
50    ],
51    deps = [
52        "//caffe2:torch",
53    ],
54)
55