CMakeLists.txt 5.3 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.
#-------------------------------------------------------------------------------

Y
yu yunfeng 已提交
7

J
jinhai 已提交
8 9 10 11
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 已提交
12
aux_source_directory(db db_files)
X
xj.lin 已提交
13
aux_source_directory(wrapper wrapper_files)
14
aux_source_directory(metrics metrics_files)
Y
yu yunfeng 已提交
15

G
groot 已提交
16 17 18 19 20 21 22 23 24
aux_source_directory(db/scheduler scheduler_files)
aux_source_directory(db/scheduler/context scheduler_context_files)
aux_source_directory(db/scheduler/task scheduler_task_files)
set(db_scheduler_files
        ${scheduler_files}
        ${scheduler_context_files}
        ${scheduler_task_files}
        )

J
jinhai 已提交
25
set(license_check_files
Y
yangwei.yao 已提交
26 27 28 29
        license/LicenseLibrary.cpp
        license/LicenseCheck.cpp
        )

Y
yu yunfeng 已提交
30
set(license_generator_files
Y
yangwei.yao 已提交
31 32
        license/LicenseGenerator.cpp
        license/LicenseLibrary.cpp
Y
yu yunfeng 已提交
33
        )
J
jinhai 已提交
34

G
groot 已提交
35
set(service_files
G
groot 已提交
36 37 38
        thrift/gen-cpp/MilvusService.cpp
        thrift/gen-cpp/milvus_constants.cpp
        thrift/gen-cpp/milvus_types.cpp
Y
yu yunfeng 已提交
39 40
        metrics/SystemInfo.cpp
        metrics/SystemInfo.h
G
groot 已提交
41 42
        server/ThreadPoolServer.cpp
        server/ThreadPoolServer.h
G
groot 已提交
43
        )
G
groot 已提交
44

G
groot 已提交
45
set(engine_files
J
jinhai 已提交
46
        ${CMAKE_CURRENT_SOURCE_DIR}/main.cpp
G
groot 已提交
47
        ${cache_files}
G
groot 已提交
48
        ${db_files}
G
groot 已提交
49
        ${db_scheduler_files}
G
groot 已提交
50
        ${wrapper_files}
Y
yu yunfeng 已提交
51 52
#        metrics/Metrics.cpp
        ${metrics_files}
G
groot 已提交
53
        )
X
xj.lin 已提交
54

Y
yu yunfeng 已提交
55
set(get_sys_info_files
Y
yangwei.yao 已提交
56
        license/GetSysInfo.cpp)
J
jinhai 已提交
57

58 59 60 61
set(s3_client_files
        storage/s3/S3ClientWrapper.cpp
        storage/s3/S3ClientWrapper.h)

G
groot 已提交
62
include_directories(/usr/include)
Z
zhiru 已提交
63
include_directories("${CUDA_TOOLKIT_ROOT_DIR}/include")
G
groot 已提交
64
include_directories(thrift/gen-cpp)
Z
zhiru 已提交
65
include_directories(/usr/include/mysql)
J
jinhai 已提交
66

Z
zhiru 已提交
67 68
set(third_party_libs
        easyloggingpp
G
groot 已提交
69 70 71
        sqlite
        thrift
        yaml-cpp
72
        libgpufaiss.a
Z
zhiru 已提交
73 74 75
        faiss
        lapack
        openblas
Z
zhiru 已提交
76 77 78
        prometheus-cpp-push
        prometheus-cpp-pull
        prometheus-cpp-core
G
groot 已提交
79 80 81 82 83
        boost_system_static
        boost_filesystem_static
        boost_serialization_static
        bzip2
        lz4
Z
zhiru 已提交
84 85 86
        snappy
        zlib
        zstd
Z
zhiru 已提交
87
        mysqlpp
Y
yu yunfeng 已提交
88
        ${CUDA_TOOLKIT_ROOT_DIR}/lib64/stubs/libnvidia-ml.so
Z
zhiru 已提交
89
        )
90 91 92
if (MEGASEARCH_WITH_ARROW STREQUAL "ON")
    set(third_party_libs ${third_party_libs} arrow)
endif() 
Z
zhiru 已提交
93

X
xj.lin 已提交
94
if (GPU_VERSION STREQUAL "ON")
Z
zhiru 已提交
95
    link_directories("${CUDA_TOOLKIT_ROOT_DIR}/lib64")
X
xj.lin 已提交
96 97 98 99 100 101
    set(engine_libs
            pthread
            libgomp.a
            libgfortran.a
            cudart
            cublas
Y
yu yunfeng 已提交
102
            ${CUDA_TOOLKIT_ROOT_DIR}/lib64/stubs/libnvidia-ml.so
X
xj.lin 已提交
103 104 105 106 107 108
            )
else()
    set(engine_libs
            pthread
            libgomp.a
            libgfortran.a
Y
yu yunfeng 已提交
109
            ${CUDA_TOOLKIT_ROOT_DIR}/lib64/stubs/libnvidia-ml.so
Y
yangwei.yao 已提交
110 111 112
            )
endif ()

113 114

if (NOT ${CMAKE_SYSTEM_PROCESSOR} MATCHES "aarch64")
W
wxyu 已提交
115 116 117 118 119 120
    set(engine_libs
	    ${engine_libs}
	    libquadmath.a
	    )
endif ()

Y
yangwei.yao 已提交
121
if (ENABLE_LICENSE STREQUAL "ON")
Z
zhiru 已提交
122 123
    link_directories("${CUDA_TOOLKIT_ROOT_DIR}/lib64/stubs")
    link_directories("${CUDA_TOOLKIT_ROOT_DIR}/lib64")
Y
yangwei.yao 已提交
124 125 126 127 128 129
    set(license_libs
            nvidia-ml
            crypto
            cudart
            cublas
            )
X
xj.lin 已提交
130 131
endif ()

G
groot 已提交
132

Y
yu yunfeng 已提交
133

G
groot 已提交
134 135
cuda_add_library(milvus_engine STATIC ${engine_files})
target_link_libraries(milvus_engine ${engine_libs} ${third_party_libs})
G
groot 已提交
136

Y
yu yunfeng 已提交
137 138
add_library(metrics STATIC ${metrics_files})

Y
yangwei.yao 已提交
139
if (ENABLE_LICENSE STREQUAL "ON")
G
groot 已提交
140 141
        add_library(license_check STATIC ${license_check_files})
        target_link_libraries(license_check ${license_libs} ${third_party_libs})
Y
yangwei.yao 已提交
142 143
endif ()

Z
zhiru 已提交
144 145 146 147 148
set(metrics_lib
        prometheus-cpp-push
        prometheus-cpp-pull
        prometheus-cpp-core
        )
Y
yu yunfeng 已提交
149

Z
zhiru 已提交
150
target_link_libraries(metrics ${metrics_lib})
Y
yu yunfeng 已提交
151

G
groot 已提交
152
set(server_libs
G
groot 已提交
153
        milvus_engine
G
groot 已提交
154
        pthread
X
xj.lin 已提交
155
        dl
Y
yu yunfeng 已提交
156
        metrics
G
groot 已提交
157
        )
G
groot 已提交
158

G
groot 已提交
159
add_executable(milvus_server
Z
zhiru 已提交
160 161 162 163 164 165 166
        ${config_files}
        ${server_files}
        ${utils_files}
        ${service_files}
        ${metrics_files}
        )

Y
yangwei.yao 已提交
167
if (ENABLE_LICENSE STREQUAL "ON")
G
groot 已提交
168
        target_link_libraries(milvus_server ${server_libs} license_check ${third_party_libs})
Y
yangwei.yao 已提交
169
else ()
G
groot 已提交
170
        target_link_libraries(milvus_server ${server_libs} ${third_party_libs})
Y
yangwei.yao 已提交
171
endif()
J
jinhai 已提交
172

Y
yangwei.yao 已提交
173
if (ENABLE_LICENSE STREQUAL "ON")
Z
zhiru 已提交
174
        add_executable(get_sys_info ${get_sys_info_files})
Y
yu yunfeng 已提交
175
        add_executable(license_generator ${license_generator_files})
Z
zhiru 已提交
176

G
groot 已提交
177
        target_link_libraries(get_sys_info ${license_libs} license_check ${third_party_libs})
Z
zhiru 已提交
178 179 180 181
        target_link_libraries(license_generator ${license_libs} ${third_party_libs})

        install(TARGETS get_sys_info DESTINATION bin)
        install(TARGETS license_generator DESTINATION bin)
Y
yangwei.yao 已提交
182
endif ()
183

G
groot 已提交
184
install(TARGETS milvus_server DESTINATION bin)
Y
yu yunfeng 已提交
185

186 187 188 189
install(FILES
        ${CMAKE_BINARY_DIR}/mysqlpp_ep-prefix/src/mysqlpp_ep/lib/${CMAKE_SHARED_LIBRARY_PREFIX}mysqlpp${CMAKE_SHARED_LIBRARY_SUFFIX}
        ${CMAKE_BINARY_DIR}/mysqlpp_ep-prefix/src/mysqlpp_ep/lib/${CMAKE_SHARED_LIBRARY_PREFIX}mysqlpp${CMAKE_SHARED_LIBRARY_SUFFIX}.3
        ${CMAKE_BINARY_DIR}/mysqlpp_ep-prefix/src/mysqlpp_ep/lib/${CMAKE_SHARED_LIBRARY_PREFIX}mysqlpp${CMAKE_SHARED_LIBRARY_SUFFIX}.3.2.4
Z
update  
zhiru 已提交
190
        DESTINATION lib) #need to copy libmysqlpp.so
Z
update  
zhiru 已提交
191

192
add_subdirectory(sdk)