• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
2
3def define_common_targets():
4    """Defines targets that should be shared between fbcode and xplat.
5
6    The directory containing this targets.bzl file should also contain both
7    TARGETS and BUCK files that call this function.
8    """
9    runtime.cxx_library(
10        name = "fast_hadamard_transform_test_impl",
11        srcs = ["fast_hadamard_transform_test_impl.cpp"],
12        exported_headers = [
13            "fast_hadamard_transform_special_unstrided_cpu.h",
14            "fast_hadamard_transform_test_impl.h",
15        ],
16        exported_deps = [
17            "//executorch/extension/llm/custom_ops/spinquant:fast_hadamard_transform",
18        ],
19        deps = [
20            "//executorch/extension/llm/custom_ops/spinquant/third-party/FFHT:dumb_fht",
21        ],
22    )
23
24    runtime.cxx_test(
25        name = "fast_hadamard_transform_test",
26        srcs = ["fast_hadamard_transform_test.cpp"],
27        deps = [
28            ":fast_hadamard_transform_test_impl",
29            "//executorch/extension/llm/custom_ops/spinquant:fast_hadamard_transform",
30        ],
31    )
32
33    runtime.cxx_test(
34        name = "op_fast_hadamard_transform_test",
35        srcs = ["op_fast_hadamard_transform_test.cpp"],
36        deps = [
37            ":fast_hadamard_transform_test_impl",
38            "//executorch/extension/llm/custom_ops:custom_ops",
39            "//executorch/kernels/test:test_util",
40            "//executorch/runtime/core/exec_aten/testing_util:tensor_util",
41        ],
42    )
43