mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 19:31:51 +08:00
68 lines
1.8 KiB
CMake
68 lines
1.8 KiB
CMake
#-------------------------------------------------------------------------------
|
|
# Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
|
|
# Unauthorized copying of this file, via any medium is strictly prohibited.
|
|
# Proprietary and confidential.
|
|
#-------------------------------------------------------------------------------
|
|
|
|
aux_source_directory(cache cache_files)
|
|
aux_source_directory(config config_files)
|
|
aux_source_directory(server server_files)
|
|
aux_source_directory(utils utils_files)
|
|
aux_source_directory(db db_files)
|
|
aux_source_directory(wrapper wrapper_files)
|
|
|
|
set(service_files
|
|
thrift/gen-cpp/VecService.cpp
|
|
thrift/gen-cpp/VectorService_constants.cpp
|
|
thrift/gen-cpp/VectorService_types.cpp
|
|
)
|
|
|
|
set(vecwise_engine_src
|
|
${CMAKE_CURRENT_SOURCE_DIR}/main.cpp
|
|
${cache_files}
|
|
${db_files}
|
|
${wrapper_files}
|
|
)
|
|
|
|
include_directories(/usr/include)
|
|
include_directories(/usr/local/cuda/include)
|
|
find_library(cuda_library cudart cublas HINTS /usr/local/cuda/lib64)
|
|
|
|
cuda_add_library(vecwise_engine STATIC ${vecwise_engine_src})
|
|
|
|
set(engine_libs
|
|
pthread
|
|
faiss
|
|
cudart
|
|
cublas
|
|
sqlite3
|
|
)
|
|
|
|
link_libraries(/usr/lib/x86_64-linux-gnu)
|
|
|
|
target_link_libraries(vecwise_engine ${engine_libs} ${cuda_library})
|
|
|
|
add_executable(vecwise_server
|
|
${config_files}
|
|
${server_files}
|
|
${utils_files}
|
|
${service_files}
|
|
${VECWISE_THIRD_PARTY_BUILD}/include/easylogging++.cc
|
|
)
|
|
|
|
set(server_libs
|
|
vecwise_engine
|
|
rocksdb
|
|
thrift
|
|
pthread
|
|
yaml-cpp
|
|
boost_system
|
|
boost_filesystem
|
|
pthread
|
|
snappy
|
|
bz2
|
|
z
|
|
)
|
|
|
|
target_link_libraries(vecwise_server ${server_libs})
|