Skip to content

Commit 29db167

Browse files
committed
Add extension/llm/custom_ops/spinquant/test to CMake
Need to run it to make sure build changes in this directory aren't breaking things ghstack-source-id: 765ef86 ghstack-comment-id: 2648964965 Pull Request resolved: #8345
1 parent e74b141 commit 29db167

File tree

5 files changed

+60
-0
lines changed

5 files changed

+60
-0
lines changed

extension/llm/custom_ops/CMakeLists.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,3 +123,8 @@ if(EXECUTORCH_BUILD_KERNELS_CUSTOM_AOT)
123123
LIBRARY DESTINATION executorch/extension/llm/custom_ops
124124
)
125125
endif()
126+
127+
add_subdirectory(spinquant/third-party/FFHT)
128+
if(BUILD_TESTING)
129+
add_subdirectory(spinquant/test)
130+
endif()
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
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+
)
Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
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)

test/run_oss_cpp_tests.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ build_executorch() {
3737
-DCMAKE_INSTALL_PREFIX=cmake-out \
3838
-DCMAKE_PREFIX_PATH="${CMAKE_PREFIX_PATH}" \
3939
-DEXECUTORCH_USE_CPP_CODE_COVERAGE=ON \
40+
-DEXECUTORCH_BUILD_KERNELS_CUSTOM=ON \
4041
-DEXECUTORCH_BUILD_KERNELS_OPTIMIZED=ON \
4142
-DEXECUTORCH_BUILD_KERNELS_QUANTIZED=ON \
4243
-DEXECUTORCH_BUILD_EXTENSION_DATA_LOADER=ON \
@@ -86,6 +87,7 @@ probe_additional_tests() {
8687
# CMakeLists.txt rules, that are buildable using build_and_run_test
8788
dirs=(
8889
examples/models/llama/tokenizer
90+
extension/llm/custom_ops
8991
extension/llm/tokenizer
9092
)
9193

test/utils/OSSTestConfig.json

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,15 @@
11
{ "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+
},
213
{
314
"directory": "extension/data_loader/test",
415
"sources": [

0 commit comments

Comments
 (0)