CMakeLists.txt 1.5 KB
Newer Older
G
groot 已提交
1 2 3 4 5
#-------------------------------------------------------------------------------
# Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
# Unauthorized copying of this file, via any medium is strictly prohibited.
# Proprietary and confidential.
#-------------------------------------------------------------------------------
6 7 8 9
aux_source_directory(${MILVUS_ENGINE_SRC}/db db_srcs)
aux_source_directory(${MILVUS_ENGINE_SRC}/config config_files)
aux_source_directory(${MILVUS_ENGINE_SRC}/cache cache_srcs)
aux_source_directory(${MILVUS_ENGINE_SRC}/wrapper wrapper_src)
G
groot 已提交
10

G
groot 已提交
11 12 13 14 15 16 17 18 19
aux_source_directory(${MILVUS_ENGINE_SRC}/db/scheduler scheduler_files)
aux_source_directory(${MILVUS_ENGINE_SRC}/db/scheduler/context scheduler_context_files)
aux_source_directory(${MILVUS_ENGINE_SRC}/db/scheduler/task scheduler_task_files)
set(db_scheduler_srcs
        ${scheduler_files}
        ${scheduler_context_files}
        ${scheduler_task_files}
        )

G
groot 已提交
20 21 22
include_directories(/usr/local/cuda/include)
link_directories("/usr/local/cuda/lib64")

23

G
groot 已提交
24 25 26

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

cuda_add_executable(db_test ${db_test_src})

set(db_libs
40
        libgpufaiss.a
G
groot 已提交
41 42 43 44 45 46
        faiss
        cudart
        cublas
        sqlite3
        boost_system
        boost_filesystem
X
Xu Peng 已提交
47
        lz4
G
groot 已提交
48 49
        )

Z
zhiru 已提交
50
target_link_libraries(db_test ${db_libs} ${unittest_libs})
51 52

install(TARGETS db_test DESTINATION bin)