diff --git a/core/CMakeLists.txt b/core/CMakeLists.txt index 760a32fdb3744421a5652a051a18007a2c3cab70..4290bc74fdffe04d91b9fcaa5fa12d893bd70288 100644 --- a/core/CMakeLists.txt +++ b/core/CMakeLists.txt @@ -178,6 +178,10 @@ if (MILVUS_WITH_PROMETHEUS) add_compile_definitions("MILVUS_WITH_PROMETHEUS") endif () +if (MILVUS_WITH_FIU) + add_compile_definitions("FIU_ENABLE") +endif () + if (CMAKE_BUILD_TYPE STREQUAL "Release") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O3 -fPIC -DELPP_THREAD_SAFE -fopenmp") if (MILVUS_GPU_VERSION) diff --git a/core/build.sh b/core/build.sh index badcc5a032fbcbb8ce63a3a6fcd3c1f0c5f1e8dd..f1a9f9406dbb324f0857d244ab7c3a89b3211ed3 100755 --- a/core/build.sh +++ b/core/build.sh @@ -17,8 +17,9 @@ WITH_MKL="OFF" FAISS_ROOT="" #FAISS root path FAISS_SOURCE="BUNDLED" WITH_PROMETHEUS="ON" +FIU_ENABLE="OFF" -while getopts "p:d:t:f:ulrcgjhxzme" arg; do +while getopts "p:d:t:f:ulrcgjhxzmei" arg; do case $arg in p) INSTALL_PREFIX=$OPTARG @@ -67,6 +68,9 @@ while getopts "p:d:t:f:ulrcgjhxzme" arg; do e) WITH_PROMETHEUS="OFF" ;; + i) + FIU_ENABLE="ON" + ;; h) # help echo " @@ -86,6 +90,7 @@ parameter: -g: build GPU version(default: OFF) -m: build with MKL(default: OFF) -e: build without prometheus(default: OFF) +-i: build FIU_ENABLE(default: OFF) -h: help usage: @@ -125,6 +130,7 @@ CMAKE_CMD="cmake \ -DMILVUS_GPU_VERSION=${GPU_VERSION} \ -DFAISS_WITH_MKL=${WITH_MKL} \ -DMILVUS_WITH_PROMETHEUS=${WITH_PROMETHEUS} \ +-DMILVUS_WITH_FIU=${FIU_ENABLE} ../" echo ${CMAKE_CMD} ${CMAKE_CMD} diff --git a/core/cmake/DefineOptions.cmake b/core/cmake/DefineOptions.cmake index 5206da864dd6d2a2127ba4f24f4615f9c6f30189..8f5a42420a8106593e6e91a5a04b62adb095d3f8 100644 --- a/core/cmake/DefineOptions.cmake +++ b/core/cmake/DefineOptions.cmake @@ -87,6 +87,8 @@ define_option(MILVUS_WITH_ZLIB "Build with zlib compression" ON) define_option(MILVUS_WITH_OPENTRACING "Build with Opentracing" ON) +define_option(MILVUS_WITH_FIU "Build with fiu" OFF) + #---------------------------------------------------------------------- set_option_category("Test and benchmark") diff --git a/core/cmake/ThirdPartyPackages.cmake b/core/cmake/ThirdPartyPackages.cmake index 8d767f82008a734f3cef357d775279611598dcc6..28ab35746ccc588fdc04414ffe67f81c43986b22 100644 --- a/core/cmake/ThirdPartyPackages.cmake +++ b/core/cmake/ThirdPartyPackages.cmake @@ -26,7 +26,8 @@ set(MILVUS_THIRDPARTY_DEPENDENCIES gperftools GRPC ZLIB - Opentracing) + Opentracing + fiu) message(STATUS "Using ${MILVUS_DEPENDENCY_SOURCE} approach to find dependencies") @@ -60,6 +61,8 @@ macro(build_dependency DEPENDENCY_NAME) build_zlib() elseif ("${DEPENDENCY_NAME}" STREQUAL "Opentracing") build_opentracing() + elseif ("${DEPENDENCY_NAME}" STREQUAL "fiu") + build_fiu() else () message(FATAL_ERROR "Unknown thirdparty dependency to build: ${DEPENDENCY_NAME}") endif () @@ -348,6 +351,12 @@ else () set(OPENTRACING_SOURCE_URL "https://github.com/opentracing/opentracing-cpp/archive/${OPENTRACING_VERSION}.tar.gz") endif () +if (DEFINED ENV{MILVUS_FIU_URL}) + set(MILVUS_FIU_URL "$ENV{MILVUS_FIU_URL}") +else () + set(FIU_SOURCE_URL "https://github.com/albertito/libfiu/archive/${FIU_VERSION}.tar.gz") +endif () + # ---------------------------------------------------------------------- # Google gtest @@ -1268,3 +1277,45 @@ if (MILVUS_WITH_OPENTRACING) get_target_property(OPENTRACING_INCLUDE_DIR opentracing INTERFACE_INCLUDE_DIRECTORIES) include_directories(SYSTEM ${OPENTRACING_INCLUDE_DIR}) endif () + +# ---------------------------------------------------------------------- +# fiu + +macro(build_fiu) + message(STATUS "Building FIU-${FIU_VERSION} from source") + set(FIU_PREFIX "${CMAKE_CURRENT_BINARY_DIR}/fiu_ep-prefix/src/fiu_ep") + set(FIU_SHARED_LIB "${FIU_PREFIX}/lib/${CMAKE_SHARED_LIBRARY_PREFIX}fiu${CMAKE_SHARED_LIBRARY_SUFFIX}") + set(FIU_INCLUDE_DIR "${FIU_PREFIX}/include") + + externalproject_add(fiu_ep + URL + ${FIU_SOURCE_URL} + ${EP_LOG_OPTIONS} + CONFIGURE_COMMAND + "" + BUILD_IN_SOURCE + 1 + BUILD_COMMAND + ${MAKE} + ${MAKE_BUILD_ARGS} + INSTALL_COMMAND + ${MAKE} + "PREFIX=${FIU_PREFIX}" + install + BUILD_BYPRODUCTS + ${FIU_SHARED_LIB} + ) + + file(MAKE_DIRECTORY "${FIU_INCLUDE_DIR}") + add_library(fiu SHARED IMPORTED) + set_target_properties(fiu + PROPERTIES IMPORTED_LOCATION "${FIU_SHARED_LIB}" + INTERFACE_INCLUDE_DIRECTORIES "${FIU_INCLUDE_DIR}") + + add_dependencies(fiu fiu_ep) +endmacro() + +resolve_dependency(fiu) + +get_target_property(FIU_INCLUDE_DIR fiu INTERFACE_INCLUDE_DIRECTORIES) +include_directories(SYSTEM ${FIU_INCLUDE_DIR}) diff --git a/core/src/CMakeLists.txt b/core/src/CMakeLists.txt index b167f862577115f2ffc7158512230fd44ff6b7e5..7709ed7f16f7874291c304bd094edd8bf08e1342 100644 --- a/core/src/CMakeLists.txt +++ b/core/src/CMakeLists.txt @@ -141,6 +141,7 @@ set(third_party_libs yaml-cpp mysqlpp zlib + fiu ${boost_lib} ) @@ -251,6 +252,9 @@ 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 + ${CMAKE_BINARY_DIR}/fiu_ep-prefix/src/fiu_ep/lib/${CMAKE_SHARED_LIBRARY_PREFIX}fiu${CMAKE_SHARED_LIBRARY_SUFFIX} + ${CMAKE_BINARY_DIR}/fiu_ep-prefix/src/fiu_ep/lib/${CMAKE_SHARED_LIBRARY_PREFIX}fiu${CMAKE_SHARED_LIBRARY_SUFFIX}.0 + ${CMAKE_BINARY_DIR}/fiu_ep-prefix/src/fiu_ep/lib/${CMAKE_SHARED_LIBRARY_PREFIX}fiu${CMAKE_SHARED_LIBRARY_SUFFIX}.1.00 DESTINATION lib) add_subdirectory(sdk) diff --git a/core/thirdparty/versions.txt b/core/thirdparty/versions.txt index e3039b47541599dc939f34a86a3e2722b68c4947..9c531c7d3983fcb666b774689e6fd425ef5eb3e6 100644 --- a/core/thirdparty/versions.txt +++ b/core/thirdparty/versions.txt @@ -10,5 +10,6 @@ GPERFTOOLS_VERSION=2.7 GRPC_VERSION=master ZLIB_VERSION=v1.2.11 OPENTRACING_VERSION=v1.5.1 +FIU_VERSION=1.00 # vim: set filetype=sh: diff --git a/core/unittest/CMakeLists.txt b/core/unittest/CMakeLists.txt index 49c0ca9f55e8e95dd60d1afa1f855c14ae9d0757..820559c06484873b4533c9ff96343d75df838f90 100644 --- a/core/unittest/CMakeLists.txt +++ b/core/unittest/CMakeLists.txt @@ -128,6 +128,7 @@ set(unittest_libs gfortran opentracing opentracing_mocktracer + fiu ) if (MILVUS_WITH_PROMETHEUS) set(unittest_libs ${unittest_libs}