File tree Expand file tree Collapse file tree 5 files changed +60
-0
lines changed Expand file tree Collapse file tree 5 files changed +60
-0
lines changed Original file line number Diff line number Diff line change @@ -123,3 +123,8 @@ if(EXECUTORCH_BUILD_KERNELS_CUSTOM_AOT)
123
123
LIBRARY DESTINATION executorch/extension/llm/custom_ops
124
124
)
125
125
endif ()
126
+
127
+ add_subdirectory (spinquant/third-party/FFHT )
128
+ if (BUILD_TESTING )
129
+ add_subdirectory (spinquant/test )
130
+ endif ()
Original file line number Diff line number Diff line change
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+ # All rights reserved.
3
+ #
4
+ # This source code is licensed under the BSD-style license found in the
5
+ # LICENSE file in the root directory of this source tree.
6
+
7
+ # @generated by test/utils/generate_gtest_cmakelists.py
8
+ #
9
+ # This file should be formatted with
10
+ # ~~~
11
+ # cmake-format -i CMakeLists.txt
12
+ # ~~~
13
+ # It should also be cmake-lint clean.
14
+ #
15
+
16
+ cmake_minimum_required (VERSION 3.19 )
17
+
18
+ set (EXECUTORCH_ROOT ${CMAKE_CURRENT_SOURCE_DIR} /../../../../.. )
19
+
20
+ include (${EXECUTORCH_ROOT} /build/Test.cmake )
21
+
22
+ set (_test_srcs
23
+ fast_hadamard_transform_test.cpp fast_hadamard_transform_test_impl.cpp
24
+ op_fast_hadamard_transform_test.cpp
25
+ )
26
+
27
+ et_cxx_test (
28
+ extension_llm_custom_ops_spinquant_test SOURCES ${_test_srcs} EXTRA_LIBS
29
+ custom_ops dumb_fht
30
+ )
Original file line number Diff line number Diff line change
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+ # All rights reserved.
3
+ #
4
+ # This source code is licensed under the BSD-style license found in the
5
+ # LICENSE file in the root directory of this source tree.
6
+
7
+ # Please this file formatted by running:
8
+ # ~~~
9
+ # cmake-format -i CMakeLists.txt
10
+ # ~~~
11
+
12
+ add_library (dumb_fht dumb_fht.c )
Original file line number Diff line number Diff line change @@ -37,6 +37,7 @@ build_executorch() {
37
37
-DCMAKE_INSTALL_PREFIX=cmake-out \
38
38
-DCMAKE_PREFIX_PATH=" ${CMAKE_PREFIX_PATH} " \
39
39
-DEXECUTORCH_USE_CPP_CODE_COVERAGE=ON \
40
+ -DEXECUTORCH_BUILD_KERNELS_CUSTOM=ON \
40
41
-DEXECUTORCH_BUILD_KERNELS_OPTIMIZED=ON \
41
42
-DEXECUTORCH_BUILD_KERNELS_QUANTIZED=ON \
42
43
-DEXECUTORCH_BUILD_EXTENSION_DATA_LOADER=ON \
@@ -86,6 +87,7 @@ probe_additional_tests() {
86
87
# CMakeLists.txt rules, that are buildable using build_and_run_test
87
88
dirs=(
88
89
examples/models/llama/tokenizer
90
+ extension/llm/custom_ops
89
91
extension/llm/tokenizer
90
92
)
91
93
Original file line number Diff line number Diff line change 1
1
{ "tests" : [
2
+ {
3
+ "directory" : " extension/llm/custom_ops/spinquant/test" ,
4
+ "sources" : [
5
+ " fast_hadamard_transform_test.cpp" ,
6
+ " fast_hadamard_transform_test_impl.cpp" ,
7
+ " op_fast_hadamard_transform_test.cpp"
8
+ ],
9
+ "additional_libs" : [
10
+ " custom_ops"
11
+ ]
12
+ },
2
13
{
3
14
"directory" : " extension/data_loader/test" ,
4
15
"sources" : [
You can’t perform that action at this time.
0 commit comments