未验证 提交 2241c977 编写于 作者: C cyberslack_lee 提交者: GitHub

[cmake] add rocksdb third party cache new (#54646)

* fix

* fix

* fix

* fix
上级 f66a705f
...@@ -16,6 +16,9 @@ include(ExternalProject) ...@@ -16,6 +16,9 @@ include(ExternalProject)
# find_package(jemalloc REQUIRED) # find_package(jemalloc REQUIRED)
set(ROCKSDB_SOURCE_DIR ${PADDLE_SOURCE_DIR}/third_party/rocksdb)
set(ROCKSDB_TAG 6.19.fb)
set(JEMALLOC_INCLUDE_DIR ${THIRD_PARTY_PATH}/install/jemalloc/include) set(JEMALLOC_INCLUDE_DIR ${THIRD_PARTY_PATH}/install/jemalloc/include)
set(JEMALLOC_LIBRARIES set(JEMALLOC_LIBRARIES
${THIRD_PARTY_PATH}/install/jemalloc/lib/libjemalloc_pic.a) ${THIRD_PARTY_PATH}/install/jemalloc/lib/libjemalloc_pic.a)
...@@ -48,6 +51,29 @@ set(ROCKSDB_CMAKE_C_FLAGS ...@@ -48,6 +51,29 @@ set(ROCKSDB_CMAKE_C_FLAGS
) )
include_directories(${ROCKSDB_INCLUDE_DIR}) include_directories(${ROCKSDB_INCLUDE_DIR})
file(GLOB ROCKSDB_SOURCE_FILE_LIST ${ROCKSDB_SOURCE_DIR})
list(LENGTH ROCKSDB_SOURCE_FILE_LIST RES_LEN)
if(RES_LEN EQUAL 0)
execute_process(
COMMAND ${GIT_EXECUTABLE} clone -b ${ROCKSDB_TAG}
"https://github.com/Thunderbrook/rocksdb" ${ROCKSDB_SOURCE_DIR})
else()
# check git tag
execute_process(
COMMAND ${GIT_EXECUTABLE} describe --abbrev=6 --always --tags
OUTPUT_VARIABLE VERSION
OUTPUT_STRIP_TRAILING_WHITESPACE ERROR_QUIET
WORKING_DIRECTORY ${ROCKSDB_SOURCE_DIR})
if(NOT ${VERSION} STREQUAL ${ROCKSDB_TAG})
file(REMOVE_RECURSE ${ROCKSDB_PREFIX_DIR})
file(REMOVE_RECURSE ${ROCKSDB_INSTALL_DIR})
message(
WARNING "rocksdb version is not ${VERSION}, checkout to ${ROCKSDB_TAG}")
execute_process(COMMAND ${GIT_EXECUTABLE} checkout ${ROCKSDB_TAG}
WORKING_DIRECTORY ${ROCKSDB_SOURCE_DIR})
endif()
endif()
set(CMAKE_CXX_LINK_EXECUTABLE set(CMAKE_CXX_LINK_EXECUTABLE
"${CMAKE_CXX_LINK_EXECUTABLE} -pthread -Wl,--no-as-needed -ldl -lrt -lz") "${CMAKE_CXX_LINK_EXECUTABLE} -pthread -Wl,--no-as-needed -ldl -lrt -lz")
if(WITH_ARM) if(WITH_ARM)
...@@ -63,8 +89,7 @@ ExternalProject_Add( ...@@ -63,8 +89,7 @@ ExternalProject_Add(
extern_rocksdb extern_rocksdb
${EXTERNAL_PROJECT_LOG_ARGS} ${EXTERNAL_PROJECT_LOG_ARGS}
PREFIX ${ROCKSDB_PREFIX_DIR} PREFIX ${ROCKSDB_PREFIX_DIR}
GIT_REPOSITORY "https://github.com/Thunderbrook/rocksdb" SOURCE_DIR ${ROCKSDB_SOURCE_DIR}
GIT_TAG 6.19.fb
UPDATE_COMMAND "" UPDATE_COMMAND ""
PATCH_COMMAND ${ROCKSDB_PATCH_COMMAND} PATCH_COMMAND ${ROCKSDB_PATCH_COMMAND}
CMAKE_ARGS -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER} CMAKE_ARGS -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER}
...@@ -81,10 +106,9 @@ ExternalProject_Add( ...@@ -81,10 +106,9 @@ ExternalProject_Add(
-DCMAKE_C_FLAGS=${CMAKE_C_FLAGS} -DCMAKE_C_FLAGS=${CMAKE_C_FLAGS}
INSTALL_COMMAND INSTALL_COMMAND
mkdir -p ${ROCKSDB_INSTALL_DIR}/lib/ && cp mkdir -p ${ROCKSDB_INSTALL_DIR}/lib/ && cp
${ROCKSDB_PREFIX_DIR}/src/extern_rocksdb/librocksdb.a ${ROCKSDB_LIBRARIES} ${ROCKSDB_PREFIX_DIR}/src/extern_rocksdb-build/librocksdb.a
&& cp -r ${ROCKSDB_PREFIX_DIR}/src/extern_rocksdb/include ${ROCKSDB_LIBRARIES} && cp -r ${ROCKSDB_SOURCE_DIR}/include
${ROCKSDB_INSTALL_DIR}/ ${ROCKSDB_INSTALL_DIR}/
BUILD_IN_SOURCE 1
BUILD_BYPRODUCTS ${ROCKSDB_LIBRARIES}) BUILD_BYPRODUCTS ${ROCKSDB_LIBRARIES})
add_library(rocksdb STATIC IMPORTED GLOBAL) add_library(rocksdb STATIC IMPORTED GLOBAL)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册