未验证 提交 34cfbe79 编写于 作者: S Sanbu 提交者: GitHub

[cmake] add dgc jemalloc third_party cache (#54392)

* add dgc jemalloc third_party cache

* fix

* fix ci

* fix ci

* fix

* fix

* fix

* Update jemalloc.cmake

* fix

* fix

* Update jemalloc.cmake

* Update jemalloc.cmake

* Update jemalloc.cmake

* Update jemalloc.cmake

* fix

* Update dgc.cmake

* fix style

* fix cmake equal to strequal

* fix

* fix

---------
Co-authored-by: Nrisemeup1 <62429225+risemeup1@users.noreply.github.com>
上级 1e323137
...@@ -12,8 +12,14 @@ ...@@ -12,8 +12,14 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
if(WIN32)
message(SEND_ERROR "The current dgc support linux only")
return()
endif()
include(ExternalProject) include(ExternalProject)
set(DGC_DOWNLOAD_DIR ${PADDLE_SOURCE_DIR}/third_party/dgc/${CMAKE_SYSTEM_NAME})
set(DGC_PREFIX_DIR "${THIRD_PARTY_PATH}/dgc") set(DGC_PREFIX_DIR "${THIRD_PARTY_PATH}/dgc")
set(DGC_SOURCES_DIR "${THIRD_PARTY_PATH}/dgc/src/extern_dgc") set(DGC_SOURCES_DIR "${THIRD_PARTY_PATH}/dgc/src/extern_dgc")
set(DGC_INSTALL_DIR "${THIRD_PARTY_PATH}/install/dgc") set(DGC_INSTALL_DIR "${THIRD_PARTY_PATH}/install/dgc")
...@@ -25,15 +31,47 @@ set(DGC_LIBRARIES ...@@ -25,15 +31,47 @@ set(DGC_LIBRARIES
CACHE FILEPATH "dgc library." FORCE) CACHE FILEPATH "dgc library." FORCE)
set(DGC_URL "https://fleet.bj.bcebos.com/dgc/collective_7369ff.tgz") set(DGC_URL "https://fleet.bj.bcebos.com/dgc/collective_7369ff.tgz")
include_directories(${DGC_INCLUDE_DIR}) include_directories(${DGC_INCLUDE_DIR})
set(DGC_CACHE_FILENAME "collective_7369ff.tgz")
set(DGC_URL_MD5 ede459281a0f979da8d84f81287369ff)
function(download_dgc)
message(
STATUS "Downloading ${DGC_URL} to ${DGC_DOWNLOAD_DIR}/${DGC_CACHE_FILENAME}"
)
# NOTE: If the version is updated, consider emptying the folder; maybe add timeout
file(
DOWNLOAD ${DGC_URL} ${DGC_DOWNLOAD_DIR}/${DGC_CACHE_FILENAME}
EXPECTED_MD5 ${DGC_URL_MD5}
STATUS ERR)
if(ERR EQUAL 0)
message(STATUS "Download ${DGC_CACHE_FILENAME} success")
else()
message(
FATAL_ERROR
"Download failed, error: ${ERR}\n You can try downloading ${DGC_CACHE_FILENAME} again"
)
endif()
endfunction()
if(EXISTS ${DGC_DOWNLOAD_DIR}/${DGC_CACHE_FILENAME})
file(MD5 ${DGC_DOWNLOAD_DIR}/${DGC_CACHE_FILENAME} DGC_MD5)
if(NOT DGC_MD5 STREQUAL DGC_URL_MD5)
download_dgc()
endif()
else()
download_dgc()
endif()
ExternalProject_Add( ExternalProject_Add(
extern_dgc extern_dgc
${EXTERNAL_PROJECT_LOG_ARGS} ${EXTERNAL_PROJECT_LOG_ARGS}
URL ${DGC_URL} URL ${DGC_DOWNLOAD_DIR}/${DGC_CACHE_FILENAME}
URL_MD5 "ede459281a0f979da8d84f81287369ff" URL_MD5 ${DGC_URL_MD5}
PREFIX "${DGC_PREFIX_DIR}" PREFIX "${DGC_PREFIX_DIR}"
CONFIGURE_COMMAND "" CONFIGURE_COMMAND ""
BUILD_COMMAND make -j${NPROC} BUILD_COMMAND make -j${NPROC}
DOWNLOAD_DIR ${DGC_DOWNLOAD_DIR}
SOURCE_DIR ${DGC_SOURCES_DIR}
INSTALL_COMMAND INSTALL_COMMAND
mkdir -p ${DGC_INSTALL_DIR}/lib/ ${DGC_INCLUDE_DIR}/dgc && cp mkdir -p ${DGC_INSTALL_DIR}/lib/ ${DGC_INCLUDE_DIR}/dgc && cp
${DGC_SOURCES_DIR}/build/lib/libdgc.a ${DGC_LIBRARIES} && cp ${DGC_SOURCES_DIR}/build/lib/libdgc.a ${DGC_LIBRARIES} && cp
......
include(ExternalProject) include(ExternalProject)
set(JEMALLOC_DOWNLOAD_DIR
${PADDLE_SOURCE_DIR}/third_party/jemalloc/${CMAKE_SYSTEM_NAME})
set(JEMALLOC_PROJECT "extern_jemalloc") set(JEMALLOC_PROJECT "extern_jemalloc")
set(JEMALLOC_BUILD ${THIRD_PARTY_PATH}/jemalloc/src/extern_jemalloc)
set(JEMALLOC_PREFIX_DIR ${THIRD_PARTY_PATH}/jemalloc)
set(JEMALLOC_URL set(JEMALLOC_URL
https://github.com/jemalloc/jemalloc/releases/download/5.1.0/jemalloc-5.1.0.tar.bz2 https://github.com/jemalloc/jemalloc/releases/download/5.1.0/jemalloc-5.1.0.tar.bz2
) )
set(JEMALLOC_BUILD ${THIRD_PARTY_PATH}/jemalloc/src/extern_jemalloc)
set(JEMALLOC_SOURCE_DIR "${THIRD_PARTY_PATH}/jemalloc")
set(JEMALLOC_INSTALL ${THIRD_PARTY_PATH}/install/jemalloc) set(JEMALLOC_INSTALL ${THIRD_PARTY_PATH}/install/jemalloc)
set(JEMALLOC_INCLUDE_DIR ${JEMALLOC_INSTALL}/include) set(JEMALLOC_INCLUDE_DIR ${JEMALLOC_INSTALL}/include)
set(JEMALLOC_DOWNLOAD_DIR "${JEMALLOC_SOURCE_DIR}/src/${JEMALLOC_PROJECT}")
set(JEMALLOC_STATIC_LIBRARIES set(JEMALLOC_STATIC_LIBRARIES
${THIRD_PARTY_PATH}/install/jemalloc/lib/libjemalloc_pic.a) ${THIRD_PARTY_PATH}/install/jemalloc/lib/libjemalloc_pic.a)
set(JEMALLOC_LIBRARIES set(JEMALLOC_LIBRARIES
${THIRD_PARTY_PATH}/install/jemalloc/lib/libjemalloc_pic.a) ${THIRD_PARTY_PATH}/install/jemalloc/lib/libjemalloc_pic.a)
set(JEMALLOC_CACHE_FILENAME "jemalloc-5.1.0.tar.bz2")
set(JEMALLOC_URL_MD5 1f47a5aff2d323c317dfa4cf23be1ce4)
function(download_jemalloc)
message(
STATUS
"Downloading ${JEMALLOC_URL} to ${JEMALLOC_DOWNLOAD_DIR}/${JEMALLOC_CACHE_FILENAME}"
)
# NOTE: If the version is updated, consider emptying the folder; maybe add timeout
file(
DOWNLOAD ${JEMALLOC_URL} ${JEMALLOC_DOWNLOAD_DIR}/${JEMALLOC_CACHE_FILENAME}
EXPECTED_MD5 ${JEMALLOC_URL_MD5}
STATUS ERR)
if(ERR EQUAL 0)
message(STATUS "Download ${JEMALLOC_CACHE_FILENAME} success")
else()
message(
FATAL_ERROR
"Download failed, error: ${ERR}\n You can try downloading ${JEMALLOC_CACHE_FILENAME} again"
)
endif()
endfunction()
if(EXISTS ${JEMALLOC_DOWNLOAD_DIR}/${JEMALLOC_CACHE_FILENAME})
file(MD5 ${JEMALLOC_DOWNLOAD_DIR}/${JEMALLOC_CACHE_FILENAME} JEMALLOC_MD5)
if(NOT JEMALLOC_MD5 STREQUAL JEMALLOC_URL_MD5)
download_jemalloc()
endif()
else()
download_jemalloc()
endif()
ExternalProject_Add( ExternalProject_Add(
extern_jemalloc extern_jemalloc
PREFIX ${JEMALLOC_SOURCE_DIR} PREFIX ${JEMALLOC_PREFIX_DIR}
URL ${JEMALLOC_URL} URL ${JEMALLOC_DOWNLOAD_DIR}/${JEMALLOC_CACHE_FILENAME}
INSTALL_DIR ${JEMALLOC_INSTALL} URL_MD5 ${JEMALLOC_URL_MD5}
DOWNLOAD_DIR "${JEMALLOC_DOWNLOAD_DIR}" #INSTALL_DIR ${JEMALLOC_INSTALL}
BUILD_COMMAND make DOWNLOAD_DIR ${JEMALLOC_DOWNLOAD_DIR}
SOURCE_DIR ${JEMALLOC_BUILD}
BUILD_IN_SOURCE 1 BUILD_IN_SOURCE 1
BUILD_COMMAND make
INSTALL_COMMAND make install INSTALL_COMMAND make install
CONFIGURE_COMMAND "${JEMALLOC_DOWNLOAD_DIR}/configure" CONFIGURE_COMMAND "${JEMALLOC_BUILD}/configure" --prefix=${JEMALLOC_INSTALL}
--prefix=${JEMALLOC_INSTALL} --disable-initial-exec-tls) --disable-initial-exec-tls)
add_library(jemalloc STATIC IMPORTED GLOBAL) add_library(jemalloc STATIC IMPORTED GLOBAL)
set_property(TARGET jemalloc PROPERTY IMPORTED_LOCATION set_property(TARGET jemalloc PROPERTY IMPORTED_LOCATION
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册