CMakeLists.txt 3.2 KB
Newer Older
Y
yu yunfeng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13


#set(CURL_LIBRARY "-lcurl")
#find_package(CURL REQUIRED)
#INCLUDE_DIRECTORIES(${CURL_INCLUDE_DIR})
#
#INCLUDE_DIRECTORIES(~/development/lib/usr/local/include)
#LINK_DIRECTORIES(~/development/lib/usr/local/lib)


include_directories(../../src)


S
starlord 已提交
14 15 16 17
aux_source_directory(${MILVUS_ENGINE_SRC}/db db_main_files)
aux_source_directory(${MILVUS_ENGINE_SRC}/db/engine db_engine_files)
aux_source_directory(${MILVUS_ENGINE_SRC}/db/insert db_insert_files)
aux_source_directory(${MILVUS_ENGINE_SRC}/db/meta db_meta_files)
S
starlord 已提交
18 19 20 21
aux_source_directory(${MILVUS_ENGINE_SRC}/config config_files)
aux_source_directory(${MILVUS_ENGINE_SRC}/cache cache_srcs)
aux_source_directory(${MILVUS_ENGINE_SRC}/wrapper wrapper_src)
aux_source_directory(${MILVUS_ENGINE_SRC}/wrapper/knowhere knowhere_src)
W
wxyu 已提交
22 23 24 25 26
aux_source_directory(${MILVUS_ENGINE_SRC}/scheduler/action scheduler_action_srcs)
aux_source_directory(${MILVUS_ENGINE_SRC}/scheduler/event scheduler_event_srcs)
aux_source_directory(${MILVUS_ENGINE_SRC}/scheduler/resource scheduler_resource_srcs)
aux_source_directory(${MILVUS_ENGINE_SRC}/scheduler/task scheduler_task_srcs)
aux_source_directory(${MILVUS_ENGINE_SRC}/scheduler scheduler_srcs)
S
starlord 已提交
27
aux_source_directory(${MILVUS_ENGINE_SRC}/src/metrics metrics_src)
G
groot 已提交
28
aux_source_directory(./ test_srcs)
Y
yu yunfeng 已提交
29

30 31 32
set(util_files
        ${MILVUS_ENGINE_SRC}/utils/ValidationUtil.cpp)

G
groot 已提交
33 34 35 36 37 38 39 40
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}
        )
Y
yu yunfeng 已提交
41 42 43 44 45 46

include_directories(/usr/include)
include_directories(../../third_party/build/include)
link_directories(../../third_party/build/lib)
include_directories(/usr/local/cuda/include)
link_directories("/usr/local/cuda/lib64")
Y
yu yunfeng 已提交
47
include_directories(../../src/metrics)
Y
yu yunfeng 已提交
48

G
groot 已提交
49
include_directories(/usr/include/mysql)
Y
yu yunfeng 已提交
50 51 52 53

set(count_test_src
        ${config_files}
        ${cache_srcs}
S
starlord 已提交
54 55 56 57
        ${db_main_files}
        ${db_engine_files}
        ${db_insert_files}
        ${db_meta_files}
G
groot 已提交
58
        ${db_scheduler_srcs}
Y
yu yunfeng 已提交
59
        ${wrapper_src}
W
wxyu 已提交
60 61 62 63 64
        ${scheduler_action_srcs}
        ${scheduler_event_srcs}
        ${scheduler_resource_srcs}
        ${scheduler_task_srcs}
        ${scheduler_srcs}
S
starlord 已提交
65
        ${knowhere_src}
66
        ${metrics_src}
G
groot 已提交
67
        ${test_srcs}
68
        ${util_files}
G
groot 已提交
69
        )
Y
yu yunfeng 已提交
70 71 72

add_executable(metrics_test ${count_test_src} ${require_files} )

S
starlord 已提交
73
set(knowhere_libs
X
MS-154  
xj.lin 已提交
74
        knowhere
S
starlord 已提交
75 76 77
        SPTAGLibStatic
        arrow
        jemalloc_pic
Y
yu yunfeng 已提交
78
        faiss
X
MS-154  
xj.lin 已提交
79 80
        openblas
        lapack
S
starlord 已提交
81 82 83 84 85
        tbb
        )

target_link_libraries(metrics_test
        ${knowhere_libs}
Y
yu yunfeng 已提交
86 87
        cudart
        cublas
88
        sqlite
Z
zhiru 已提交
89 90
        boost_system_static
        boost_filesystem_static
Y
yu yunfeng 已提交
91 92 93 94 95
        lz4
        metrics
        gtest
        pthread
        z
G
groot 已提交
96
        mysqlpp
Z
zhiru 已提交
97
        ${unittest_libs}
98
        )
99
if(${BUILD_FAISS_WITH_MKL} STREQUAL "ON")
Z
zhiru 已提交
100
    target_link_libraries(metrics_test ${MKL_LIBS} ${MKL_LIBS})
Z
fix  
zhiru 已提交
101 102 103 104
else()
    target_link_libraries(metrics_test
            lapack
            openblas)
Z
zhiru 已提交
105
endif()
106 107

install(TARGETS metrics_test DESTINATION bin)