Skip to content

Commit 83f8c40

Browse files
committed
Merge branch 'develop' of https://github.com/paddlepaddle/paddle into fix_avg_pool_trt_bug
test=develop
2 parents b969116 + 2f27c04 commit 83f8c40

File tree

152 files changed

+3748
-1219
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

152 files changed

+3748
-1219
lines changed

CMakeLists.txt

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,11 @@ message(STATUS "C compiler: ${CMAKE_C_COMPILER}, version: "
2626
"${CMAKE_C_COMPILER_ID} ${CMAKE_C_COMPILER_VERSION}")
2727
if(WIN32)
2828
set(CMAKE_STATIC_LIBRARY_PREFIX lib)
29+
add_definitions("/DGOOGLE_GLOG_DLL_DECL=")
30+
set(CMAKE_C_FLAGS_DEBUG "${CMAKE_C_FLAGS_DEBUG} /bigobj /MTd")
31+
set(CMAKE_C_FLAGS_RELEASE "${CMAKE_C_FLAGS_RELEASE} /bigobj /MT")
32+
set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /bigobj /MTd")
33+
set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /bigobj /MT")
2934
endif(WIN32)
3035

3136
if(NOT CMAKE_CROSSCOMPILING)
@@ -66,6 +71,8 @@ option(WITH_ARM_FP16 "Use half precision support on armv8.2-a cpu" OFF)
6671
option(WITH_CONTRIB "Compile the third-party contributation" OFF)
6772
option(REPLACE_ENFORCE_GLOG "Replace PADDLE_ENFORCE with glog/CHECK for better debug." OFF)
6873
option(WITH_ANAKIN "Compile with Anakin library" OFF)
74+
option(ANAKIN_BUILD_FAT_BIN "Build anakin cuda fat-bin lib for all device plantform, ignored when WITH_ANAKIN=OFF" OFF)
75+
option(ANAKIN_BUILD_CROSS_PLANTFORM "Build anakin lib for any nvidia device plantform. ignored when WITH_ANAKIN=OFF" ON)
6976
option(WITH_GRPC "Use grpc as the default rpc framework" ${WITH_DISTRIBUTE})
7077
option(WITH_BRPC_RDMA "Use brpc rdma as the rpc protocal" OFF)
7178
option(ON_INFER "Turn on inference optimization." OFF)
@@ -308,7 +315,6 @@ endif()
308315

309316
if (ON_INFER)
310317
message(STATUS "On inference mode, will take place some specific optimization.")
311-
add_definitions(-DPADDLE_ON_INFERENCE)
312318
else()
313319
#TODO(luotao), combine this warning with `make inference_lib_dist` command.
314320
message(WARNING "On inference mode, will take place some specific optimization. Turn on the ON_INFER flag when building inference_lib only.")

cmake/configure.cmake

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -218,3 +218,7 @@ endif(WITH_GRPC)
218218
if(WITH_BRPC_RDMA)
219219
add_definitions(-DPADDLE_WITH_BRPC_RDMA)
220220
endif(WITH_BRPC_RDMA)
221+
222+
if(ON_INFER)
223+
add_definitions(-DPADDLE_ON_INFERENCE)
224+
endif(ON_INFER)

cmake/cuda.cmake

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,9 @@ list(APPEND EXTERNAL_LIBS ${CUDA_LIBRARIES} ${CUDA_rt_LIBRARY})
157157
if(NOT WITH_DSO)
158158
# TODO(panyx0718): CUPTI only allows DSO?
159159
list(APPEND EXTERNAL_LIBS ${CUDNN_LIBRARY} ${CUPTI_LIBRARY} ${CUDA_CUBLAS_LIBRARIES} ${CUDA_curand_LIBRARY} ${NCCL_LIBRARY})
160+
if(WIN32)
161+
set_property(GLOBAL PROPERTY CUDA_MODULES ${CUDNN_LIBRARY} ${CUDA_CUBLAS_LIBRARIES} ${CUDA_curand_LIBRARY})
162+
endif(WIN32)
160163
endif(NOT WITH_DSO)
161164

162165
# setting nvcc arch flags
@@ -196,10 +199,12 @@ elseif(CMAKE_BUILD_TYPE STREQUAL "MinSizeRel")
196199
list(APPEND CUDA_NVCC_FLAGS ${CMAKE_CXX_FLAGS_RELEASE})
197200
endif()
198201
else(NOT WIN32)
199-
if(CMAKE_BUILD_TYPE STREQUAL "Release")
202+
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
203+
list(APPEND CUDA_NVCC_FLAGS "-g -G")
204+
elseif(CMAKE_BUILD_TYPE STREQUAL "Release")
200205
list(APPEND CUDA_NVCC_FLAGS "-O3 -DNDEBUG")
201206
else()
202-
message(FATAL "Windows only support Release build now. Please set visual studio build type to Release, x64 build.")
207+
message(FATAL "Windows only support Release or Debug build now. Please set visual studio build type to Release/Debug, x64 build.")
203208
endif()
204209
endif(NOT WIN32)
205210

cmake/cudnn.cmake

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,12 @@ if(NOT WITH_GPU)
22
return()
33
endif()
44

5-
set(CUDNN_ROOT "/usr" CACHE PATH "CUDNN ROOT")
5+
if(WIN32)
6+
set(CUDNN_ROOT ${CUDA_TOOLKIT_ROOT_DIR})
7+
else(WIN32)
8+
set(CUDNN_ROOT "/usr" CACHE PATH "CUDNN ROOT")
9+
endif(WIN32)
10+
611
find_path(CUDNN_INCLUDE_DIR cudnn.h
712
PATHS ${CUDNN_ROOT} ${CUDNN_ROOT}/include
813
$ENV{CUDNN_ROOT} $ENV{CUDNN_ROOT}/include ${CUDA_TOOLKIT_INCLUDE}

cmake/external/anakin.cmake

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -58,19 +58,21 @@ ExternalProject_Add(
5858
-DPROTOBUF_ROOT=${THIRD_PARTY_PATH}/install/protobuf
5959
-DMKLML_ROOT=${THIRD_PARTY_PATH}/install/mklml
6060
-DENABLE_OP_TIMER=${ANAKIN_ENABLE_OP_TIMER}
61+
-DBUILD_FAT_BIN=${ANAKIN_BUILD_FAT_BIN}
62+
-DBUILD_CROSS_PLANTFORM=${ANAKIN_BUILD_CROSS_PLANTFORM}
6163
${EXTERNAL_OPTIONAL_ARGS}
6264
CMAKE_CACHE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=${ANAKIN_INSTALL_DIR}
6365
)
6466

6567
message(STATUS "Anakin for inference is enabled")
6668
message(STATUS "Anakin is set INCLUDE:${ANAKIN_INCLUDE} LIBRARY:${ANAKIN_LIBRARY}")
67-
69+
add_dependencies(extern_anakin protobuf mklml)
6870
add_library(anakin_shared SHARED IMPORTED GLOBAL)
6971
set_property(TARGET anakin_shared PROPERTY IMPORTED_LOCATION ${ANAKIN_SHARED_LIB})
70-
add_dependencies(anakin_shared extern_anakin protobuf mklml)
72+
add_dependencies(anakin_shared extern_anakin)
7173

7274
add_library(anakin_saber SHARED IMPORTED GLOBAL)
7375
set_property(TARGET anakin_saber PROPERTY IMPORTED_LOCATION ${ANAKIN_SABER_LIB})
74-
add_dependencies(anakin_saber extern_anakin protobuf mklml)
76+
add_dependencies(anakin_saber extern_anakin)
7577

7678
list(APPEND external_project_dependencies anakin_shared anakin_saber)

cmake/external/boost.cmake

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -28,34 +28,28 @@ if((NOT DEFINED BOOST_TAR) OR (NOT DEFINED BOOST_URL))
2828
set(BOOST_TAR "boost_1_41_0" CACHE STRING "" FORCE)
2929
set(BOOST_URL "http://paddlepaddledeps.cdn.bcebos.com/${BOOST_TAR}.tar.gz" CACHE STRING "" FORCE)
3030
endif()
31-
IF (WIN32)
32-
MESSAGE(WARNING, "In windows, boost can not be downloaded automaticlly, please build it manually and put it at " ${THIRD_PARTY_PATH}install/boost)
33-
else()
34-
MESSAGE(STATUS "BOOST_TAR: ${BOOST_TAR}, BOOST_URL: ${BOOST_URL}")
35-
ENDIF(WIN32)
31+
32+
MESSAGE(STATUS "BOOST_TAR: ${BOOST_TAR}, BOOST_URL: ${BOOST_URL}")
3633

3734
set(BOOST_SOURCES_DIR ${THIRD_PARTY_PATH}/boost)
3835
set(BOOST_DOWNLOAD_DIR "${BOOST_SOURCES_DIR}/src/${BOOST_PROJECT}")
39-
set(BOOST_INCLUDE_DIR "${BOOST_DOWNLOAD_DIR}/${BOOST_TAR}" CACHE PATH "boost include directory." FORCE)
40-
set_directory_properties(PROPERTIES CLEAN_NO_CUSTOM 1)
4136

37+
set(BOOST_INCLUDE_DIR "${BOOST_DOWNLOAD_DIR}" CACHE PATH "boost include directory." FORCE)
38+
set_directory_properties(PROPERTIES CLEAN_NO_CUSTOM 1)
4239
include_directories(${BOOST_INCLUDE_DIR})
4340

44-
if (NOT WIN32)
4541
ExternalProject_Add(
4642
${BOOST_PROJECT}
4743
${EXTERNAL_PROJECT_LOG_ARGS}
4844
DOWNLOAD_DIR ${BOOST_DOWNLOAD_DIR}
49-
DOWNLOAD_COMMAND wget --no-check-certificate ${BOOST_URL} -c -q -O ${BOOST_TAR}.tar.gz
50-
&& tar zxf ${BOOST_TAR}.tar.gz
45+
URL ${BOOST_URL}
5146
DOWNLOAD_NO_PROGRESS 1
5247
PREFIX ${BOOST_SOURCES_DIR}
5348
CONFIGURE_COMMAND ""
5449
BUILD_COMMAND ""
5550
INSTALL_COMMAND ""
5651
UPDATE_COMMAND ""
57-
)
58-
endif(NOT WIN32)
52+
)
5953

6054
if (${CMAKE_VERSION} VERSION_LESS "3.3.0" OR NOT WIN32)
6155
set(dummyfile ${CMAKE_CURRENT_BINARY_DIR}/boost_dummy.c)

cmake/external/gflags.cmake

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,12 @@ ExternalProject_Add(
3535
CMAKE_ARGS -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER}
3636
-DCMAKE_C_COMPILER=${CMAKE_C_COMPILER}
3737
-DCMAKE_CXX_FLAGS=${CMAKE_CXX_FLAGS}
38+
-DCMAKE_CXX_FLAGS_RELEASE=${CMAKE_CXX_FLAGS_RELEASE}
39+
-DCMAKE_CXX_FLAGS_DEBUG=${CMAKE_CXX_FLAGS_DEBUG}
3840
-DCMAKE_C_FLAGS=${CMAKE_C_FLAGS}
41+
-DCMAKE_C_FLAGS_DEBUG=${CMAKE_C_FLAGS_DEBUG}
42+
-DCMAKE_C_FLAGS_RELEASE=${CMAKE_C_FLAGS_RELEASE}
43+
-DBUILD_STATIC_LIBS=ON
3944
-DCMAKE_INSTALL_PREFIX=${GFLAGS_INSTALL_DIR}
4045
-DCMAKE_POSITION_INDEPENDENT_CODE=ON
4146
-DBUILD_TESTING=OFF
@@ -48,8 +53,8 @@ ExternalProject_Add(
4853
IF(WIN32)
4954
IF(NOT EXISTS "${GFLAGS_INSTALL_DIR}/lib/libgflags.lib")
5055
add_custom_command(TARGET extern_gflags POST_BUILD
51-
COMMAND cmake -E rename ${GFLAGS_INSTALL_DIR}/lib/gflags_static.lib ${GFLAGS_INSTALL_DIR}/lib/libgflags.lib
52-
)
56+
COMMAND cmake -E copy ${GFLAGS_INSTALL_DIR}/lib/gflags_static.lib ${GFLAGS_INSTALL_DIR}/lib/libgflags.lib
57+
)
5358
ENDIF()
5459
ENDIF(WIN32)
5560
ADD_LIBRARY(gflags STATIC IMPORTED GLOBAL)

cmake/external/glog.cmake

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,11 @@ ExternalProject_Add(
4646
CMAKE_ARGS -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER}
4747
-DCMAKE_C_COMPILER=${CMAKE_C_COMPILER}
4848
-DCMAKE_CXX_FLAGS=${CMAKE_CXX_FLAGS}
49+
-DCMAKE_CXX_FLAGS_RELEASE=${CMAKE_CXX_FLAGS_RELEASE}
50+
-DCMAKE_CXX_FLAGS_DEBUG=${CMAKE_CXX_FLAGS_DEBUG}
4951
-DCMAKE_C_FLAGS=${CMAKE_C_FLAGS}
52+
-DCMAKE_C_FLAGS_DEBUG=${CMAKE_C_FLAGS_DEBUG}
53+
-DCMAKE_C_FLAGS_RELEASE=${CMAKE_C_FLAGS_RELEASE}
5054
-DCMAKE_INSTALL_PREFIX=${GLOG_INSTALL_DIR}
5155
-DCMAKE_INSTALL_LIBDIR=${GLOG_INSTALL_DIR}/lib
5256
-DCMAKE_POSITION_INDEPENDENT_CODE=ON
@@ -63,7 +67,7 @@ ExternalProject_Add(
6367
IF(WIN32)
6468
IF(NOT EXISTS "${GLOG_INSTALL_DIR}/lib/libglog.lib")
6569
add_custom_command(TARGET extern_glog POST_BUILD
66-
COMMAND cmake -E rename ${GLOG_INSTALL_DIR}/lib/glog.lib ${GLOG_INSTALL_DIR}/lib/libglog.lib
70+
COMMAND cmake -E copy ${GLOG_INSTALL_DIR}/lib/glog.lib ${GLOG_INSTALL_DIR}/lib/libglog.lib
6771
)
6872
ENDIF()
6973
ENDIF(WIN32)

cmake/external/openblas.cmake

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,12 +17,8 @@ IF(USE_EIGEN_FOR_BLAS)
1717
ENDIF(USE_EIGEN_FOR_BLAS)
1818

1919
INCLUDE(cblas)
20-
# IF(WIN32 AND NOT ${CBLAS_FOUND})
21-
22-
2320

2421
IF(NOT ${CBLAS_FOUND})
25-
2622
INCLUDE(ExternalProject)
2723

2824
SET(CBLAS_SOURCES_DIR ${THIRD_PARTY_PATH}/openblas)
@@ -34,6 +30,7 @@ IF(NOT ${CBLAS_FOUND})
3430
CACHE FILEPATH "openblas library." FORCE)
3531

3632
ADD_DEFINITIONS(-DPADDLE_USE_OPENBLAS)
33+
3734
IF (WIN32)
3835
SET(CBLAS_FOUND true)
3936
MESSAGE(WARNING, "In windows, openblas only support msvc build, please build it manually and put it at " ${CBLAS_INSTALL_DIR})

cmake/external/protobuf.cmake

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,6 @@ endmacro()
140140
set(PROTOBUF_ROOT "" CACHE PATH "Folder contains protobuf")
141141
IF (WIN32)
142142
SET(PROTOBUF_ROOT ${THIRD_PARTY_PATH}/install/protobuf)
143-
MESSAGE(WARNING, "In windows, protobuf only support msvc build, please build it manually and put it at " ${PROTOBUF_ROOT})
144143
ENDIF(WIN32)
145144

146145
if (NOT "${PROTOBUF_ROOT}" STREQUAL "")
@@ -188,13 +187,20 @@ FUNCTION(build_protobuf TARGET_NAME BUILD_FOR_HOST)
188187
SET(OPTIONAL_ARGS
189188
"-DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER}"
190189
"-DCMAKE_C_COMPILER=${CMAKE_C_COMPILER}"
191-
"-DCMAKE_CXX_FLAGS=${CMAKE_CXX_FLAGS}"
192190
"-DCMAKE_C_FLAGS=${CMAKE_C_FLAGS}"
191+
"-DCMAKE_C_FLAGS_DEBUG=${CMAKE_C_FLAGS_DEBUG}"
192+
"-DCMAKE_C_FLAGS_RELEASE=${CMAKE_C_FLAGS_RELEASE}"
193+
"-DCMAKE_CXX_FLAGS=${CMAKE_CXX_FLAGS}"
194+
"-DCMAKE_CXX_FLAGS_RELEASE=${CMAKE_CXX_FLAGS_RELEASE}"
195+
"-DCMAKE_CXX_FLAGS_DEBUG=${CMAKE_CXX_FLAGS_DEBUG}"
193196
"-Dprotobuf_WITH_ZLIB=ON"
194197
"-DZLIB_ROOT:FILEPATH=${ZLIB_ROOT}"
195198
${EXTERNAL_OPTIONAL_ARGS})
196199
SET(OPTIONAL_CACHE_ARGS "-DZLIB_ROOT:STRING=${ZLIB_ROOT}")
197200
ENDIF()
201+
IF(WIN32)
202+
SET(OPTIONAL_ARGS ${OPTIONAL_ARGS} "-DCMAKE_GENERATOR_PLATFORM=x64")
203+
ENDIF()
198204

199205
SET(PROTOBUF_REPO "https://github.com/google/protobuf.git")
200206
SET(PROTOBUF_TAG "9f75c5aa851cd877fb0d93ccc31b8567a6706546")

0 commit comments

Comments
 (0)