CMakeLists.txt 3.1 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
cmake_minimum_required(VERSION 3.12)
J
jinhai 已提交
8

X
xj.lin 已提交
9 10 11 12 13 14 15
project(vecwise_engine LANGUAGES CUDA CXX)

find_package(CUDA)
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -Xcompiler -fPIC -std=c++11 -D_FORCE_INLINES -arch sm_60 --expt-extended-lambda")
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -O0 -g")
message("CUDA_TOOLKIT_ROOT_DIR=${CUDA_TOOLKIT_ROOT_DIR}")
message("CUDA_NVCC_FLAGS=${CUDA_NVCC_FLAGS}")
J
jinhai 已提交
16

X
xj.lin 已提交
17 18 19 20
if (GPU_VERSION STREQUAL "ON")
    add_definitions("-DGPU_VERSION")
endif ()

J
jinhai 已提交
21 22 23 24 25 26 27 28 29 30 31 32 33 34
set(CMAKE_CXX_STANDARD 14)

if(CMAKE_SYSTEM_PROCESSOR MATCHES "(x86)|(X86)|(amd64)|(AMD64)")
    message("building vecwise_engine on x86 architecture")
    set(VECWISE_BUILD_ARCH x86_64)
elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "(ppc)")
    message("building vecwise_engine on ppc architecture")
    set(VECWISE_BUILD_ARCH ppc64le)
else()
    message("unknown processor type")
    message("CMAKE_SYSTEM_PROCESSOR=${CMAKE_SYSTEM_PROCESSOR}")
    set(VECWISE_BUILD_ARCH unknown)
endif()

Y
yu yunfeng 已提交
35 36 37 38 39 40 41 42 43 44
if(DEFINED UNIX)
    message("building vecwise on Unix")
    set(VECWISE_BUILD_SYSTEM macos)
elseif(DEFINED APPLE)
    message("building vecwise on MacOS")
    set(VECWISE_BUILD_SYSTEM unix)
else()
    message("unknown OS")
    set(VECWISE_BUILD_SYSTEM unknown)
endif ()
J
jinhai 已提交
45 46

if(CMAKE_BUILD_TYPE STREQUAL "Release")
J
jinhai 已提交
47
    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O3 -fPIC -DELPP_THREAD_SAFE")
Y
yu yunfeng 已提交
48 49 50 51
    if (GPU_VERSION STREQUAL "ON")
        set(ENABLE_LICENSE "ON")
        add_definitions("-DENABLE_LICENSE")
    endif ()
J
jinhai 已提交
52
else()
J
jinhai 已提交
53
    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O0 -g -fPIC -DELPP_THREAD_SAFE")
J
jinhai 已提交
54 55
endif()

Y
yu yunfeng 已提交
56
set(CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmake/Modules/" ${CMAKE_MODULE_PATH})
J
jinhai 已提交
57 58 59

set(VECWISE_ENGINE_INCLUDE ${PROJECT_SOURCE_DIR}/include)
set(VECWISE_ENGINE_SRC ${PROJECT_SOURCE_DIR}/src)
J
jinhai 已提交
60 61
set(VECWISE_THIRD_PARTY ${CMAKE_CURRENT_SOURCE_DIR}/third_party)
set(VECWISE_THIRD_PARTY_BUILD ${CMAKE_CURRENT_SOURCE_DIR}/third_party/build)
J
jinhai 已提交
62 63 64 65 66

add_compile_definitions(PROFILER=${PROFILER})

include_directories(${VECWISE_ENGINE_INCLUDE})
include_directories(${VECWISE_ENGINE_SRC})
J
jinhai 已提交
67 68
include_directories(${VECWISE_THIRD_PARTY_BUILD}/include)

G
groot 已提交
69
link_directories(${CMAKE_CURRRENT_BINARY_DIR})
J
jinhai 已提交
70
link_directories(${VECWISE_THIRD_PARTY_BUILD}/lib)
Z
zilliz 已提交
71
link_directories(${VECWISE_THIRD_PARTY_BUILD}/lib64)
72 73
#execute_process(COMMAND bash build.sh
#                WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/third_party)
J
jinhai 已提交
74 75

add_subdirectory(src)
Y
yu yunfeng 已提交
76
add_subdirectory(unittest)
Y
yu yunfeng 已提交
77

J
jinhai 已提交
78 79

if (BUILD_UNIT_TEST)
J
jinhai 已提交
80
    add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/unittest)
J
jinhai 已提交
81
endif(BUILD_UNIT_TEST)
Y
yu yunfeng 已提交
82

G
groot 已提交
83
add_custom_target(Clean-All COMMAND ${CMAKE_BUILD_TOOL} clean)
84 85 86

#install
install(FILES
X
xj.lin 已提交
87 88
        scripts/start_server.sh
        scripts/stop_server.sh
89 90 91 92 93 94 95
        DESTINATION
        scripts)
install(FILES
        conf/server_config.yaml
        conf/vecwise_engine_log.conf
        DESTINATION
        conf)