mirror of https://github.com/milvus-io/milvus.git
18 lines
486 B
CMake
18 lines
486 B
CMake
include_directories("${CUDA_TOOLKIT_ROOT_DIR}/include")
|
|
link_directories("${CUDA_TOOLKIT_ROOT_DIR}/lib64")
|
|
|
|
aux_source_directory(${MILVUS_ENGINE_SRC}/wrapper/knowhere knowhere_src)
|
|
|
|
set(helper
|
|
utils.cpp)
|
|
|
|
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) |