修改cmake

zjhellofss · Jun 22, 2024 · e43d937 · e43d937
1 parent f9439c8
commit e43d937
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -13,7 +13,7 @@ aux_source_directory(kuiper/source/op/kernels DIR_KERNEL)
 aux_source_directory(kuiper/source/sampler DIR_SAMPLE)
 
 set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${PROJECT_SOURCE_DIR}/lib)
-
+find_package(GTest REQUIRED)
 add_library(llama SHARED ${DIR_TENSOR} ${DIR_BASE} ${DIR_OP} ${DIR_MODEL} ${DIR_KERNEL} ${DIR_SAMPLE})
 target_link_libraries(llama sentencepiece glog::glog  gtest gtest_main pthread)
 target_link_libraries(llama ${ARMADILLO_LIBRARIES})