milvus/cpp/unittest/faiss_wrapper/CMakeLists.txt
xj.lin 3fd083c987 add wrapper and unittest file
Former-commit-id: 9b115de1c9d84ed4c790980793fec28169ca7837
2019-04-16 11:05:38 +08:00

27 lines
912 B
CMake

#-------------------------------------------------------------------------------
# Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
# Unauthorized copying of this file, via any medium is strictly prohibited.
# Proprietary and confidential.
#-------------------------------------------------------------------------------
include_directories(../../src)
aux_source_directory(../../src/wrapper wrapper_src)
# Make sure that your call to link_directories takes place before your call to the relevant add_executable.
include_directories(/usr/local/cuda/include)
link_directories("/usr/local/cuda/lib64")
set(wrapper_test_src
${unittest_srcs}
${wrapper_src}
wrapper_test.cpp)
add_executable(wrapper_test ${wrapper_test_src})
set(faiss_libs
faiss
cudart
cublas
)
target_link_libraries(wrapper_test ${unittest_libs} ${faiss_libs})