CMakeLists.txt 1.5 KB
Newer Older
J
jinhai 已提交
1 2 3 4 5 6
#-------------------------------------------------------------------------------
# Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
# Unauthorized copying of this file, via any medium is strictly prohibited.
# Proprietary and confidential.
#-------------------------------------------------------------------------------

J
jinhai 已提交
7 8 9 10
aux_source_directory(cache cache_files)
aux_source_directory(config config_files)
aux_source_directory(server server_files)
aux_source_directory(utils utils_files)
G
groot 已提交
11
aux_source_directory(db db_files)
X
xj.lin 已提交
12
aux_source_directory(wrapper wrapper_files)
G
groot 已提交
13

G
groot 已提交
14 15 16 17 18
set(service_files
        thrift/gen-cpp/VecService.cpp
        thrift/gen-cpp/VectorService_constants.cpp
        thrift/gen-cpp/VectorService_types.cpp)

J
jinhai 已提交
19 20
set(vecwise_engine_src
        ${CMAKE_CURRENT_SOURCE_DIR}/main.cpp
G
groot 已提交
21
        ${cache_files}
G
groot 已提交
22
        ${db_files}
X
xj.lin 已提交
23 24 25 26
        ${wrapper_files})

include_directories(/usr/local/cuda/include)
find_library(cuda_library cudart cublas HINTS /usr/local/cuda/lib64)
J
jinhai 已提交
27

J
jinhai 已提交
28
add_library(vecwise_engine STATIC ${vecwise_engine_src})
G
groot 已提交
29

X
xj.lin 已提交
30
cuda_add_executable(vecwise_server
G
groot 已提交
31 32 33
        ${config_files}
        ${server_files}
        ${utils_files}
G
groot 已提交
34
        ${service_files}
X
xj.lin 已提交
35
        ${wrapper_files}
G
groot 已提交
36
        ${service_files}
G
groot 已提交
37
        ${VECWISE_THIRD_PARTY_BUILD}/include/easylogging++.cc
G
groot 已提交
38 39 40 41 42 43 44
        )

set(dependency_libs
        vecwise_engine
        yaml-cpp
        boost_system
        boost_filesystem
G
groot 已提交
45 46
        thrift
        pthread
G
groot 已提交
47 48
        faiss
        vecwise_engine
X
Xu Peng 已提交
49
        sqlite3
G
groot 已提交
50
        )
G
groot 已提交
51
target_link_libraries(vecwise_server ${dependency_libs} ${cuda_library})