修改cmake

zjhellofss · Jun 22, 2024 · 8b3f144 · 8b3f144
1 parent 0d36f12
commit 8b3f144
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -15,7 +15,7 @@ aux_source_directory(kuiper/source/sampler DIR_SAMPLE)
 set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${PROJECT_SOURCE_DIR}/lib)
 
 add_library(llama SHARED ${DIR_TENSOR} ${DIR_BASE} ${DIR_OP} ${DIR_MODEL} ${DIR_KERNEL} ${DIR_SAMPLE})
-target_link_libraries(llama sentencepiece glog::glog)
+target_link_libraries(llama sentencepiece glog::glog  gtest gtest_main pthread)
 target_link_libraries(llama ${ARMADILLO_LIBRARIES})
 target_include_directories(llama PUBLIC ${glog_INCLUDE_DIR})
 target_include_directories(llama PUBLIC ${PROJECT_SOURCE_DIR}/kuiper/include)
@@ -24,5 +24,4 @@ target_include_directories(llama PUBLIC ${Armadillo_INCLUDE_DIR})
 add_executable(llama_infer main.cpp)
 target_link_directories(llama_infer PUBLIC ${PROJECT_SOURCE_DIR}/lib)
 target_link_libraries(llama_infer llama)
-
 add_subdirectory(test)