CMakeLists.txt 1.2 KB
Newer Older
G
groot 已提交
1 2 3 4 5 6 7 8
#-------------------------------------------------------------------------------
# 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/db db_srcs)
G
groot 已提交
9 10
aux_source_directory(../../src/config config_files)
aux_source_directory(../../src/cache cache_srcs)
G
groot 已提交
11 12 13 14 15
aux_source_directory(../../src/wrapper wrapper_src)

include_directories(/usr/local/cuda/include)
link_directories("/usr/local/cuda/lib64")

G
groot 已提交
16 17 18 19 20
set(require_files
        ../../src/server/ServerConfig.cpp
        ../../src/utils/CommonUtil.cpp
        ../../src/utils/TimeRecorder.cpp
        )
G
groot 已提交
21 22 23

set(db_test_src
    ${unittest_srcs}
G
groot 已提交
24 25
    ${config_files}
    ${cache_srcs}
G
groot 已提交
26 27
    ${db_srcs}
    ${wrapper_src}
G
groot 已提交
28
    ${require_files}
X
Xu Peng 已提交
29
    utils.cpp
X
Xu Peng 已提交
30
    db_tests.cpp
X
Xu Peng 已提交
31
    meta_tests.cpp)
G
groot 已提交
32 33 34 35 36 37 38 39 40 41

cuda_add_executable(db_test ${db_test_src})

set(db_libs
        faiss
        cudart
        cublas
        sqlite3
        boost_system
        boost_filesystem
X
Xu Peng 已提交
42
        lz4
G
groot 已提交
43 44
        )

Z
zhiru 已提交
45
target_link_libraries(db_test ${db_libs} ${unittest_libs})