CMakeLists.txt 1.9 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
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)
X
xj.lin 已提交
9
#aux_source_directory(${MILVUS_ENGINE_SRC}/wrapper wrapper_src)
S
starlord 已提交
10
aux_source_directory(${MILVUS_ENGINE_SRC}/wrapper/knowhere knowhere_src)
G
groot 已提交
11
aux_source_directory(./ test_srcs)
G
groot 已提交
12

G
groot 已提交
13 14 15 16 17 18 19 20 21
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 已提交
22 23 24
include_directories(/usr/local/cuda/include)
link_directories("/usr/local/cuda/lib64")

Z
zhiru 已提交
25
include_directories(/usr/include/mysql)
G
groot 已提交
26

Z
update  
zhiru 已提交
27 28
#add_definitions(-DBOOST_ERROR_CODE_HEADER_ONLY)

G
groot 已提交
29
set(db_test_src
Z
zhiru 已提交
30
    #${unittest_srcs}
G
groot 已提交
31 32
    ${config_files}
    ${cache_srcs}
G
groot 已提交
33
    ${db_srcs}
G
groot 已提交
34
    ${db_scheduler_srcs}
G
groot 已提交
35
    ${wrapper_src}
S
starlord 已提交
36
    ${knowhere_src}
G
groot 已提交
37
    ${require_files}
S
starlord 已提交
38 39
    ${test_srcs}
        )
G
groot 已提交
40 41 42 43

cuda_add_executable(db_test ${db_test_src})

set(db_libs
44
        sqlite
Z
zhiru 已提交
45
        boost_filesystem_static
Z
update  
zhiru 已提交
46
        boost_system_static
X
Xu Peng 已提交
47
        lz4
Z
zhiru 已提交
48
        mysqlpp
G
groot 已提交
49 50
        )

S
starlord 已提交
51 52 53 54 55 56 57 58 59
set(knowhere_libs
        knowhere
        SPTAGLibStatic
        arrow
        jemalloc_pic
        faiss
        openblas
        lapack
        tbb
S
starlord 已提交
60 61
        cudart
        cublas
S
starlord 已提交
62 63
        )

Z
update  
zhiru 已提交
64
target_link_libraries(db_test ${knowhere_libs} ${db_libs} ${unittest_libs})
65

Z
zhiru 已提交
66
install(TARGETS db_test DESTINATION bin)