2019-05-28 05:34:47 +00:00
|
|
|
|
|
|
|
|
|
|
|
#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)
|
|
|
|
|
|
|
|
|
|
|
|
|
2019-07-08 01:42:54 +00:00
|
|
|
aux_source_directory(${MILVUS_ENGINE_SRC}/db db_srcs)
|
|
|
|
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)
|
|
|
|
aux_source_directory(${MILVUS_ENGINE_SRC}/src/metrics metrics_src)
|
2019-06-30 08:19:04 +00:00
|
|
|
aux_source_directory(./ test_srcs)
|
2019-05-28 09:50:20 +00:00
|
|
|
|
2019-07-22 02:44:24 +00:00
|
|
|
set(util_files
|
|
|
|
${MILVUS_ENGINE_SRC}/utils/ValidationUtil.cpp)
|
|
|
|
|
2019-06-20 08:05:44 +00:00
|
|
|
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}
|
|
|
|
)
|
2019-05-28 05:34:47 +00:00
|
|
|
|
|
|
|
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")
|
2019-05-28 09:50:20 +00:00
|
|
|
include_directories(../../src/metrics)
|
2019-05-28 05:34:47 +00:00
|
|
|
|
2019-06-30 08:19:04 +00:00
|
|
|
include_directories(/usr/include/mysql)
|
2019-05-28 05:34:47 +00:00
|
|
|
|
|
|
|
set(count_test_src
|
|
|
|
${config_files}
|
|
|
|
${cache_srcs}
|
|
|
|
${db_srcs}
|
2019-06-04 11:32:43 +00:00
|
|
|
${db_scheduler_srcs}
|
2019-05-28 05:34:47 +00:00
|
|
|
${wrapper_src}
|
2019-07-08 01:42:54 +00:00
|
|
|
${knowhere_src}
|
2019-06-19 04:03:13 +00:00
|
|
|
${metrics_src}
|
2019-06-30 08:19:04 +00:00
|
|
|
${test_srcs}
|
2019-07-22 02:44:24 +00:00
|
|
|
${util_files}
|
2019-06-30 08:19:04 +00:00
|
|
|
)
|
2019-05-28 05:34:47 +00:00
|
|
|
|
|
|
|
add_executable(metrics_test ${count_test_src} ${require_files} )
|
|
|
|
|
2019-07-08 01:42:54 +00:00
|
|
|
set(knowhere_libs
|
2019-07-04 06:47:23 +00:00
|
|
|
knowhere
|
2019-07-08 01:42:54 +00:00
|
|
|
SPTAGLibStatic
|
|
|
|
arrow
|
|
|
|
jemalloc_pic
|
2019-05-28 05:34:47 +00:00
|
|
|
faiss
|
2019-07-04 06:47:23 +00:00
|
|
|
openblas
|
|
|
|
lapack
|
2019-07-08 01:42:54 +00:00
|
|
|
tbb
|
|
|
|
)
|
|
|
|
|
|
|
|
target_link_libraries(metrics_test
|
|
|
|
${knowhere_libs}
|
2019-05-28 05:34:47 +00:00
|
|
|
cudart
|
|
|
|
cublas
|
2019-07-09 09:07:52 +00:00
|
|
|
sqlite
|
2019-07-09 09:31:02 +00:00
|
|
|
boost_system_static
|
|
|
|
boost_filesystem_static
|
2019-05-28 05:34:47 +00:00
|
|
|
lz4
|
|
|
|
metrics
|
|
|
|
gtest
|
|
|
|
pthread
|
|
|
|
z
|
2019-06-30 08:19:04 +00:00
|
|
|
mysqlpp
|
2019-05-29 08:14:15 +00:00
|
|
|
${unittest_libs}
|
2019-06-19 07:46:14 +00:00
|
|
|
)
|
2019-07-21 12:46:24 +00:00
|
|
|
if(${BUILD_FAISS_WITH_MKL} STREQUAL "ON")
|
2019-07-17 04:50:37 +00:00
|
|
|
target_link_libraries(metrics_test ${MKL_LIBS} ${MKL_LIBS})
|
2019-07-22 03:16:14 +00:00
|
|
|
else()
|
|
|
|
target_link_libraries(metrics_test
|
|
|
|
lapack
|
|
|
|
openblas)
|
2019-07-17 04:50:37 +00:00
|
|
|
endif()
|
2019-06-19 07:46:14 +00:00
|
|
|
|
|
|
|
install(TARGETS metrics_test DESTINATION bin)
|