Merge pull request #1 from PaddlePaddle/develop
update to merevert-15207-remove_op_handle_lock_and_fix_var
lujun
6 years ago
committed by
GitHub
commit
6daad7c990
@ -0,0 +1,63 @@
|
|||||||
|
# Tries to find Gperftools.
|
||||||
|
#
|
||||||
|
# Usage of this module as follows:
|
||||||
|
#
|
||||||
|
# find_package(Gperftools)
|
||||||
|
#
|
||||||
|
# Variables used by this module, they can change the default behaviour and need
|
||||||
|
# to be set before calling find_package:
|
||||||
|
#
|
||||||
|
# Gperftools_ROOT_DIR Set this variable to the root installation of
|
||||||
|
# Gperftools if the module has problems finding
|
||||||
|
# the proper installation path.
|
||||||
|
#
|
||||||
|
# Variables defined by this module:
|
||||||
|
#
|
||||||
|
# GPERFTOOLS_FOUND System has Gperftools libs/headers
|
||||||
|
# GPERFTOOLS_LIBRARIES The Gperftools libraries (tcmalloc & profiler)
|
||||||
|
# GPERFTOOLS_INCLUDE_DIR The location of Gperftools headers
|
||||||
|
|
||||||
|
find_library(GPERFTOOLS_TCMALLOC
|
||||||
|
NAMES tcmalloc
|
||||||
|
HINTS ${Gperftools_ROOT_DIR}/lib)
|
||||||
|
|
||||||
|
find_library(GPERFTOOLS_PROFILER
|
||||||
|
NAMES profiler
|
||||||
|
HINTS ${Gperftools_ROOT_DIR}/lib)
|
||||||
|
|
||||||
|
find_library(GPERFTOOLS_TCMALLOC_AND_PROFILER
|
||||||
|
NAMES tcmalloc_and_profiler
|
||||||
|
HINTS ${Gperftools_ROOT_DIR}/lib)
|
||||||
|
|
||||||
|
find_path(GPERFTOOLS_INCLUDE_DIR
|
||||||
|
NAMES gperftools/heap-profiler.h
|
||||||
|
HINTS ${Gperftools_ROOT_DIR}/include)
|
||||||
|
|
||||||
|
set(GPERFTOOLS_LIBRARIES ${GPERFTOOLS_TCMALLOC_AND_PROFILER})
|
||||||
|
|
||||||
|
include(FindPackageHandleStandardArgs)
|
||||||
|
find_package_handle_standard_args(
|
||||||
|
Gperftools
|
||||||
|
DEFAULT_MSG
|
||||||
|
GPERFTOOLS_LIBRARIES
|
||||||
|
GPERFTOOLS_INCLUDE_DIR)
|
||||||
|
|
||||||
|
mark_as_advanced(
|
||||||
|
Gperftools_ROOT_DIR
|
||||||
|
GPERFTOOLS_TCMALLOC
|
||||||
|
GPERFTOOLS_PROFILER
|
||||||
|
GPERFTOOLS_TCMALLOC_AND_PROFILER
|
||||||
|
GPERFTOOLS_LIBRARIES
|
||||||
|
GPERFTOOLS_INCLUDE_DIR)
|
||||||
|
|
||||||
|
# create IMPORTED targets
|
||||||
|
if (Gperftools_FOUND AND NOT TARGET gperftools::tcmalloc)
|
||||||
|
add_library(gperftools::tcmalloc UNKNOWN IMPORTED)
|
||||||
|
set_target_properties(gperftools::tcmalloc PROPERTIES
|
||||||
|
IMPORTED_LOCATION ${GPERFTOOLS_TCMALLOC}
|
||||||
|
INTERFACE_INCLUDE_DIRECTORIES "${GPERFTOOLS_INCLUDE_DIR}")
|
||||||
|
add_library(gperftools::profiler UNKNOWN IMPORTED)
|
||||||
|
set_target_properties(gperftools::profiler PROPERTIES
|
||||||
|
IMPORTED_LOCATION ${GPERFTOOLS_PROFILER}
|
||||||
|
INTERFACE_INCLUDE_DIRECTORIES "${GPERFTOOLS_INCLUDE_DIR}")
|
||||||
|
endif()
|
||||||
@ -0,0 +1,31 @@
|
|||||||
|
include(ExternalProject)
|
||||||
|
|
||||||
|
set(DLPACK_SOURCE_DIR ${THIRD_PARTY_PATH}/dlpack)
|
||||||
|
set(DLPACK_INCLUDE_DIR ${DLPACK_SOURCE_DIR}/src/extern_dlpack/include)
|
||||||
|
|
||||||
|
include_directories(${DLPACK_INCLUDE_DIR})
|
||||||
|
|
||||||
|
ExternalProject_Add(
|
||||||
|
extern_dlpack
|
||||||
|
${EXTERNAL_PROJECT_LOG_ARGS}
|
||||||
|
GIT_REPOSITORY "https://github.com/dmlc/dlpack.git"
|
||||||
|
GIT_TAG "v0.2"
|
||||||
|
PREFIX ${DLPACK_SOURCE_DIR}
|
||||||
|
UPDATE_COMMAND ""
|
||||||
|
CONFIGURE_COMMAND ""
|
||||||
|
BUILD_COMMAND ""
|
||||||
|
INSTALL_COMMAND ""
|
||||||
|
TEST_COMMAND ""
|
||||||
|
)
|
||||||
|
|
||||||
|
if(${CMAKE_VERSION} VERSION_LESS "3.3.0")
|
||||||
|
set(dummyfile ${CMAKE_CURRENT_BINARY_DIR}/dlpack_dummy.c)
|
||||||
|
file(WRITE ${dummyfile} "const char *dummy = \"${dummyfile}\";")
|
||||||
|
add_library(dlpack STATIC ${dummyfile})
|
||||||
|
else()
|
||||||
|
add_library(dlpack INTERFACE)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
add_dependencies(dlpack extern_dlpack)
|
||||||
|
|
||||||
|
LIST(APPEND externl_project_dependencies dlpack)
|
||||||
@ -0,0 +1,48 @@
|
|||||||
|
# Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
#
|
||||||
|
|
||||||
|
IF(MOBILE_INFERENCE)
|
||||||
|
return()
|
||||||
|
ENDIF()
|
||||||
|
|
||||||
|
include (ExternalProject)
|
||||||
|
|
||||||
|
# NOTE: gzstream is needed when linking with ctr reader.
|
||||||
|
|
||||||
|
SET(GZSTREAM_SOURCES_DIR ${THIRD_PARTY_PATH}/gzstream)
|
||||||
|
SET(GZSTREAM_INSTALL_DIR ${THIRD_PARTY_PATH}/install/gzstream)
|
||||||
|
SET(GZSTREAM_INCLUDE_DIR "${GZSTREAM_INSTALL_DIR}/include/" CACHE PATH "gzstream include directory." FORCE)
|
||||||
|
|
||||||
|
ExternalProject_Add(
|
||||||
|
extern_gzstream
|
||||||
|
DEPENDS zlib
|
||||||
|
GIT_REPOSITORY "https://github.com/jacquesqiao/gzstream.git"
|
||||||
|
GIT_TAG ""
|
||||||
|
PREFIX ${GZSTREAM_SOURCES_DIR}
|
||||||
|
UPDATE_COMMAND ""
|
||||||
|
CONFIGURE_COMMAND ""
|
||||||
|
BUILD_IN_SOURCE 1
|
||||||
|
BUILD_COMMAND make EXTERN_CPPFLAGS="-I${THIRD_PARTY_PATH}/install/zlib/include" EXTERM_LDFLAGS="-L${THIRD_PARTY_PATH}/install/zlib/lib" -j8
|
||||||
|
INSTALL_COMMAND mkdir -p ${GZSTREAM_INSTALL_DIR}/lib/ && mkdir -p ${GZSTREAM_INSTALL_DIR}/include/
|
||||||
|
&& cp ${GZSTREAM_SOURCES_DIR}/src/extern_gzstream/libgzstream.a ${GZSTREAM_INSTALL_DIR}/lib
|
||||||
|
&& cp -r ${GZSTREAM_SOURCES_DIR}/src/extern_gzstream/gzstream.h ${GZSTREAM_INSTALL_DIR}/include
|
||||||
|
)
|
||||||
|
|
||||||
|
ADD_LIBRARY(gzstream STATIC IMPORTED GLOBAL)
|
||||||
|
SET_PROPERTY(TARGET gzstream PROPERTY IMPORTED_LOCATION
|
||||||
|
"${GZSTREAM_INSTALL_DIR}/lib/libgzstream.a")
|
||||||
|
|
||||||
|
include_directories(${GZSTREAM_INCLUDE_DIR})
|
||||||
|
ADD_DEPENDENCIES(gzstream extern_gzstream zlib)
|
||||||
@ -0,0 +1,78 @@
|
|||||||
|
# Copyright (c) 2017 PaddlePaddle Authors. All Rights Reserved.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
IF(NOT ${WITH_LIBMCT})
|
||||||
|
return()
|
||||||
|
ENDIF(NOT ${WITH_LIBMCT})
|
||||||
|
|
||||||
|
IF(WIN32 OR APPLE)
|
||||||
|
MESSAGE(WARNING
|
||||||
|
"Windows or Mac is not supported with LIBMCT in Paddle yet."
|
||||||
|
"Force WITH_LIBMCT=OFF")
|
||||||
|
SET(WITH_LIBMCT OFF CACHE STRING "Disable LIBMCT package in Windows and MacOS" FORCE)
|
||||||
|
return()
|
||||||
|
ENDIF()
|
||||||
|
|
||||||
|
INCLUDE(ExternalProject)
|
||||||
|
|
||||||
|
SET(LIBMCT_PROJECT "extern_libmct")
|
||||||
|
IF((NOT DEFINED LIBMCT_VER) OR (NOT DEFINED LIBMCT_URL))
|
||||||
|
MESSAGE(STATUS "use pre defined download url")
|
||||||
|
SET(LIBMCT_VER "0.1.0" CACHE STRING "" FORCE)
|
||||||
|
SET(LIBMCT_NAME "libmct" CACHE STRING "" FORCE)
|
||||||
|
SET(LIBMCT_URL "https://raw.githubusercontent.com/PaddlePaddle/Fleet/release/${LIBMCT_VER}/${LIBMCT_NAME}.tar.gz" CACHE STRING "" FORCE)
|
||||||
|
ENDIF()
|
||||||
|
MESSAGE(STATUS "LIBMCT_NAME: ${LIBMCT_NAME}, LIBMCT_URL: ${LIBMCT_URL}")
|
||||||
|
SET(LIBMCT_SOURCE_DIR "${THIRD_PARTY_PATH}/libmct")
|
||||||
|
SET(LIBMCT_DOWNLOAD_DIR "${LIBMCT_SOURCE_DIR}/src/${LIBMCT_PROJECT}")
|
||||||
|
SET(LIBMCT_DST_DIR "libmct")
|
||||||
|
SET(LIBMCT_INSTALL_ROOT "${THIRD_PARTY_PATH}/install")
|
||||||
|
SET(LIBMCT_INSTALL_DIR ${LIBMCT_INSTALL_ROOT}/${LIBMCT_DST_DIR})
|
||||||
|
SET(LIBMCT_ROOT ${LIBMCT_INSTALL_DIR})
|
||||||
|
SET(LIBMCT_INC_DIR ${LIBMCT_ROOT}/include)
|
||||||
|
SET(CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_RPATH}" "${LIBMCT_ROOT}/lib")
|
||||||
|
|
||||||
|
INCLUDE_DIRECTORIES(${LIBMCT_INC_DIR})
|
||||||
|
|
||||||
|
FILE(WRITE ${LIBMCT_DOWNLOAD_DIR}/CMakeLists.txt
|
||||||
|
"PROJECT(LIBMCT)\n"
|
||||||
|
"cmake_minimum_required(VERSION 3.0)\n"
|
||||||
|
"install(DIRECTORY ${LIBMCT_NAME}/include ${LIBMCT_NAME}/lib \n"
|
||||||
|
" DESTINATION ${LIBMCT_DST_DIR})\n")
|
||||||
|
|
||||||
|
ExternalProject_Add(
|
||||||
|
${LIBMCT_PROJECT}
|
||||||
|
${EXTERNAL_PROJECT_LOG_ARGS}
|
||||||
|
PREFIX ${LIBMCT_SOURCE_DIR}
|
||||||
|
DOWNLOAD_DIR ${LIBMCT_DOWNLOAD_DIR}
|
||||||
|
DOWNLOAD_COMMAND wget --no-check-certificate ${LIBMCT_URL} -c -q -O ${LIBMCT_NAME}.tar.gz
|
||||||
|
&& tar zxvf ${LIBMCT_NAME}.tar.gz
|
||||||
|
DOWNLOAD_NO_PROGRESS 1
|
||||||
|
UPDATE_COMMAND ""
|
||||||
|
CMAKE_ARGS -DCMAKE_INSTALL_PREFIX=${LIBMCT_INSTALL_ROOT}
|
||||||
|
CMAKE_CACHE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=${LIBMCT_INSTALL_ROOT}
|
||||||
|
)
|
||||||
|
|
||||||
|
if (${CMAKE_VERSION} VERSION_LESS "3.3.0" OR NOT WIN32)
|
||||||
|
set(dummyfile ${CMAKE_CURRENT_BINARY_DIR}/boost_dummy.c)
|
||||||
|
file(WRITE ${dummyfile} "const char *dummy = \"${dummyfile}\";")
|
||||||
|
add_library(libmct STATIC ${dummyfile})
|
||||||
|
else()
|
||||||
|
add_library(libmct INTERFACE)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
#ADD_LIBRARY(libmct SHARED IMPORTED GLOBAL)
|
||||||
|
ADD_DEPENDENCIES(libmct ${LIBMCT_PROJECT})
|
||||||
|
LIST(APPEND external_project_dependencies libmct)
|
||||||
|
|
||||||
@ -0,0 +1,77 @@
|
|||||||
|
# Copyright (c) 2017 PaddlePaddle Authors. All Rights Reserved.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
IF(NOT ${WITH_PSLIB})
|
||||||
|
return()
|
||||||
|
ENDIF(NOT ${WITH_PSLIB})
|
||||||
|
|
||||||
|
IF(WIN32 OR APPLE)
|
||||||
|
MESSAGE(WARNING
|
||||||
|
"Windows or Mac is not supported with PSLIB in Paddle yet."
|
||||||
|
"Force WITH_PSLIB=OFF")
|
||||||
|
SET(WITH_PSLIB OFF CACHE STRING "Disable PSLIB package in Windows and MacOS" FORCE)
|
||||||
|
return()
|
||||||
|
ENDIF()
|
||||||
|
|
||||||
|
INCLUDE(ExternalProject)
|
||||||
|
|
||||||
|
SET(PSLIB_PROJECT "extern_pslib")
|
||||||
|
IF((NOT DEFINED PSLIB_VER) OR (NOT DEFINED PSLIB_URL))
|
||||||
|
MESSAGE(STATUS "use pre defined download url")
|
||||||
|
SET(PSLIB_VER "0.1.0" CACHE STRING "" FORCE)
|
||||||
|
SET(PSLIB_NAME "pslib" CACHE STRING "" FORCE)
|
||||||
|
SET(PSLIB_URL "https://raw.githubusercontent.com/PaddlePaddle/Fleet/release/${PSLIB_VER}/${PSLIB_NAME}.tar.gz" CACHE STRING "" FORCE)
|
||||||
|
ENDIF()
|
||||||
|
MESSAGE(STATUS "PSLIB_NAME: ${PSLIB_NAME}, PSLIB_URL: ${PSLIB_URL}")
|
||||||
|
SET(PSLIB_SOURCE_DIR "${THIRD_PARTY_PATH}/pslib")
|
||||||
|
SET(PSLIB_DOWNLOAD_DIR "${PSLIB_SOURCE_DIR}/src/${PSLIB_PROJECT}")
|
||||||
|
SET(PSLIB_DST_DIR "pslib")
|
||||||
|
SET(PSLIB_INSTALL_ROOT "${THIRD_PARTY_PATH}/install")
|
||||||
|
SET(PSLIB_INSTALL_DIR ${PSLIB_INSTALL_ROOT}/${PSLIB_DST_DIR})
|
||||||
|
SET(PSLIB_ROOT ${PSLIB_INSTALL_DIR})
|
||||||
|
SET(PSLIB_INC_DIR ${PSLIB_ROOT}/include)
|
||||||
|
SET(PSLIB_LIB_DIR ${PSLIB_ROOT}/lib)
|
||||||
|
SET(PSLIB_LIB ${PSLIB_LIB_DIR}/libps.so)
|
||||||
|
SET(PSLIB_IOMP_LIB ${PSLIB_LIB_DIR}/libiomp5.so) #todo what is this
|
||||||
|
SET(CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_RPATH}" "${PSLIB_ROOT}/lib")
|
||||||
|
|
||||||
|
INCLUDE_DIRECTORIES(${PSLIB_INC_DIR})
|
||||||
|
|
||||||
|
FILE(WRITE ${PSLIB_DOWNLOAD_DIR}/CMakeLists.txt
|
||||||
|
"PROJECT(PSLIB)\n"
|
||||||
|
"cmake_minimum_required(VERSION 3.0)\n"
|
||||||
|
"install(DIRECTORY ${PSLIB_NAME}/include ${PSLIB_NAME}/lib \n"
|
||||||
|
" DESTINATION ${PSLIB_DST_DIR})\n")
|
||||||
|
|
||||||
|
ExternalProject_Add(
|
||||||
|
${PSLIB_PROJECT}
|
||||||
|
${EXTERNAL_PROJECT_LOG_ARGS}
|
||||||
|
PREFIX ${PSLIB_SOURCE_DIR}
|
||||||
|
DOWNLOAD_DIR ${PSLIB_DOWNLOAD_DIR}
|
||||||
|
DOWNLOAD_COMMAND wget --no-check-certificate ${PSLIB_URL} -c -q -O ${PSLIB_NAME}.tar.gz
|
||||||
|
&& tar zxvf ${PSLIB_NAME}.tar.gz
|
||||||
|
DOWNLOAD_NO_PROGRESS 1
|
||||||
|
UPDATE_COMMAND ""
|
||||||
|
CMAKE_ARGS -DCMAKE_INSTALL_PREFIX=${PSLIB_INSTALL_ROOT}
|
||||||
|
CMAKE_CACHE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=${PSLIB_INSTALL_ROOT}
|
||||||
|
)
|
||||||
|
|
||||||
|
ADD_LIBRARY(pslib SHARED IMPORTED GLOBAL)
|
||||||
|
SET_PROPERTY(TARGET pslib PROPERTY IMPORTED_LOCATION ${PSLIB_LIB})
|
||||||
|
ADD_DEPENDENCIES(pslib ${PSLIB_PROJECT})
|
||||||
|
LIST(APPEND external_project_dependencies pslib)
|
||||||
|
|
||||||
|
IF(WITH_C_API)
|
||||||
|
INSTALL(FILES ${PSLIB_LIB} ${PSLIB_IOMP_LIB} DESTINATION lib)
|
||||||
|
ENDIF()
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue