1load("@fbsource//xplat/executorch/kernels/test:util.bzl", "define_supported_features_lib", "op_test") 2 3def define_common_targets(): 4 define_supported_features_lib() 5 6 op_test("op_quantize_test", kernel_name = "quantized") 7 op_test("op_dequantize_test", kernel_name = "quantized") 8 op_test("op_choose_qparams_test", kernel_name = "quantized") 9 op_test("op_add_test", kernel_name = "quantized", deps = [ 10 "//executorch/kernels/quantized/cpu:op_dequantize", 11 "//executorch/kernels/quantized/cpu:op_quantize", 12 "//executorch/kernels/quantized/cpu:op_add", 13 "//executorch/kernels/quantized:generated_lib_headers", 14 "//executorch/kernels/portable:generated_lib_headers", 15 "//executorch/kernels/portable/cpu:op_add", 16 "//executorch/runtime/core/exec_aten/testing_util:tensor_util", 17 ]) 18 op_test("op_embedding_test", kernel_name = "quantized", deps = [ 19 "//executorch/kernels/quantized/cpu:op_dequantize", 20 "//executorch/kernels/quantized/cpu:op_quantize", 21 "//executorch/kernels/quantized/cpu:op_add", 22 "//executorch/kernels/quantized/cpu:op_embedding", 23 "//executorch/kernels/quantized:generated_lib_headers", 24 "//executorch/kernels/portable:generated_lib_headers", 25 "//executorch/kernels/portable/cpu:op_embedding", 26 "//executorch/runtime/core/exec_aten/testing_util:tensor_util", 27 ]) 28 op_test("op_embedding2b_test", kernel_name = "quantized") 29 op_test("op_embedding4b_test", kernel_name = "quantized") 30 op_test("op_mixed_mm_test", kernel_name = "quantized", deps = [ 31 "//executorch/kernels/quantized/cpu:op_mixed_mm", 32 "//executorch/kernels/quantized:generated_lib_headers", 33 "//executorch/kernels/portable:generated_lib_headers", 34 "//executorch/runtime/core/exec_aten/testing_util:tensor_util", 35 ]) 36 op_test("op_mixed_linear_test", kernel_name = "quantized", deps = [ 37 "//executorch/kernels/quantized/cpu:op_mixed_linear", 38 "//executorch/kernels/quantized:generated_lib_headers", 39 "//executorch/kernels/portable:generated_lib_headers", 40 "//executorch/runtime/core/exec_aten/testing_util:tensor_util", 41 ]) 42