mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 19:31:51 +08:00
23 lines
717 B
CMake
23 lines
717 B
CMake
include_directories("${CUDA_TOOLKIT_ROOT_DIR}/include")
|
|
link_directories("${CUDA_TOOLKIT_ROOT_DIR}/lib64")
|
|
|
|
set(knowhere_src
|
|
${MILVUS_ENGINE_SRC}/wrapper/knowhere/data_transfer.cpp
|
|
${MILVUS_ENGINE_SRC}/wrapper/knowhere/vec_impl.cpp
|
|
${MILVUS_ENGINE_SRC}/wrapper/knowhere/vec_index.cpp)
|
|
|
|
set(helper
|
|
utils.cpp
|
|
${MILVUS_ENGINE_SRC}/utils/easylogging++.cc
|
|
${MILVUS_ENGINE_SRC}/utils/easylogging++.h)
|
|
|
|
set(knowhere_libs
|
|
knowhere
|
|
cudart
|
|
cublas
|
|
)
|
|
|
|
add_executable(knowhere_test knowhere_test.cpp ${knowhere_src} ${helper})
|
|
target_link_libraries(knowhere_test ${knowhere_libs} ${unittest_libs})
|
|
|
|
install(TARGETS knowhere_test DESTINATION unittest) |