You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
174 lines
6.9 KiB
174 lines
6.9 KiB
# Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.
|
|
#
|
|
# 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
|
|
|
|
cmake_minimum_required(VERSION 3.0)
|
|
set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CMAKE_CURRENT_SOURCE_DIR}/cmake")
|
|
set(PROJ_ROOT ${CMAKE_CURRENT_SOURCE_DIR})
|
|
set(PROJ_BINARY_ROOT ${CMAKE_CURRENT_BINARY_DIR})
|
|
|
|
include(system)
|
|
|
|
project(paddle CXX C Go)
|
|
|
|
find_package(Sphinx)
|
|
if(NOT CMAKE_CROSSCOMPILING)
|
|
find_package(CUDA QUIET)
|
|
endif(NOT CMAKE_CROSSCOMPILING)
|
|
find_package(Git REQUIRED)
|
|
find_package(Threads REQUIRED)
|
|
if(NOT ANDROID)
|
|
find_package(Boost QUIET)
|
|
endif()
|
|
|
|
include(simd)
|
|
|
|
################################ Configurations #######################################
|
|
option(WITH_GPU "Compile PaddlePaddle with NVIDIA GPU" ${CUDA_FOUND})
|
|
option(WITH_AVX "Compile PaddlePaddle with AVX intrinsics" ${AVX_FOUND})
|
|
option(WITH_MKLDNN "Compile PaddlePaddle with mkl-dnn support." ${AVX_FOUND})
|
|
option(WITH_MKLML "Compile PaddlePaddle with mklml package." ${AVX_FOUND})
|
|
option(WITH_DSO "Compile PaddlePaddle with dynamic linked CUDA" ON)
|
|
option(WITH_TESTING "Compile PaddlePaddle with unit testing" ON)
|
|
option(WITH_SWIG_PY "Compile PaddlePaddle with inference api" ON)
|
|
option(WITH_STYLE_CHECK "Compile PaddlePaddle with style check" ON)
|
|
option(WITH_PYTHON "Compile PaddlePaddle with python interpreter" ON)
|
|
option(WITH_DOUBLE "Compile PaddlePaddle with double precision" OFF)
|
|
option(WITH_RDMA "Compile PaddlePaddle with RDMA support" OFF)
|
|
option(WITH_TIMER "Compile PaddlePaddle with stats timer" OFF)
|
|
option(WITH_PROFILER "Compile PaddlePaddle with GPU profiler" OFF)
|
|
option(WITH_DOC "Compile PaddlePaddle with documentation" OFF)
|
|
option(WITH_COVERAGE "Compile PaddlePaddle with code coverage" OFF)
|
|
option(COVERALLS_UPLOAD "Package code coverage data to coveralls" OFF)
|
|
option(ON_TRAVIS "Exclude special unit test on Travis CI" OFF)
|
|
option(WITH_C_API "Compile PaddlePaddle with C-API(Prediction)" OFF)
|
|
option(WITH_GOLANG "Compile PaddlePaddle with GOLANG" OFF)
|
|
option(GLIDE_INSTALL "Download and install go dependencies " ON)
|
|
option(USE_NNPACK "Compile PaddlePaddle with NNPACK library" OFF)
|
|
|
|
# CMAKE_BUILD_TYPE
|
|
if(NOT CMAKE_BUILD_TYPE)
|
|
set(CMAKE_BUILD_TYPE "RelWithDebInfo" CACHE STRING
|
|
"Choose the type of build, options are: Debug Release RelWithDebInfo MinSizeRel"
|
|
FORCE)
|
|
endif()
|
|
|
|
if(ANDROID)
|
|
if(${CMAKE_SYSTEM_VERSION} VERSION_LESS "21")
|
|
message(FATAL_ERROR "Unsupport standalone toolchains with Android API level lower than 21")
|
|
endif()
|
|
|
|
set(WITH_GPU OFF CACHE STRING
|
|
"Disable GPU when cross-compiling for Android" FORCE)
|
|
set(WITH_AVX OFF CACHE STRING
|
|
"Disable AVX when cross-compiling for Android" FORCE)
|
|
set(WITH_PYTHON OFF CACHE STRING
|
|
"Disable PYTHON when cross-compiling for Android" FORCE)
|
|
set(WITH_RDMA OFF CACHE STRING
|
|
"Disable RDMA when cross-compiling for Android" FORCE)
|
|
set(WITH_MKLDNN OFF CACHE STRING
|
|
"Disable MKLDNN when cross-compiling for Android" FORCE)
|
|
set(WITH_MKLML OFF CACHE STRING
|
|
"Disable MKLML package when cross-compiling for Android" FORCE)
|
|
endif(ANDROID)
|
|
|
|
set(THIRD_PARTY_PATH "${CMAKE_BINARY_DIR}/third_party" CACHE STRING
|
|
"A path setting third party libraries download & build directories.")
|
|
|
|
if (WITH_C_API AND WITH_PYTHON)
|
|
message(WARNING "It is suggest not embedded a python interpreter in Paddle "
|
|
"when using C-API. It will give an unpredictable behavior when using a "
|
|
"different Python interpreter from compiling.")
|
|
endif()
|
|
|
|
########################################################################################
|
|
|
|
include(external/mklml) # download mklml package
|
|
include(external/zlib) # download, build, install zlib
|
|
include(external/gflags) # download, build, install gflags
|
|
include(external/glog) # download, build, install glog
|
|
include(external/gtest) # download, build, install gtest
|
|
include(external/protobuf) # download, build, install protobuf
|
|
include(external/python) # download, build, install python
|
|
include(external/openblas) # download, build, install openblas
|
|
include(external/mkldnn) # download, build, install mkldnn
|
|
include(external/swig) # download, build, install swig
|
|
include(external/warpctc) # download, build, install warpctc
|
|
include(external/any) # download libn::any
|
|
include(external/eigen) # download eigen3
|
|
include(external/pybind11) # download pybind11
|
|
|
|
include(cudnn) # set cudnn libraries, must before configure
|
|
include(configure) # add paddle env configuration
|
|
include(generic) # simplify cmake module
|
|
include(package) # set paddle packages
|
|
include(cpplint) # set paddle c++ style
|
|
include(ccache) # set ccache for compilation
|
|
include(util) # set unittest and link libs
|
|
include(rdma) # set rdma libraries
|
|
include(flags) # set paddle compile flags
|
|
include(version) # set PADDLE_VERSION
|
|
include(coveralls) # set code coverage
|
|
|
|
|
|
include_directories("${PROJ_ROOT}")
|
|
include_directories("${PROJ_ROOT}/paddle/cuda/include")
|
|
include_directories("${CMAKE_CURRENT_BINARY_DIR}/proto")
|
|
include_directories("${CMAKE_CURRENT_BINARY_DIR}/go/pserver/client/c")
|
|
include_directories(${Boost_INCLUDE_DIRS})
|
|
|
|
set(EXTERNAL_LIBS
|
|
${GFLAGS_LIBRARIES}
|
|
${GLOG_LIBRARIES}
|
|
${CBLAS_LIBRARIES}
|
|
${PROTOBUF_LIBRARY}
|
|
${ZLIB_LIBRARIES}
|
|
${PYTHON_LIBRARIES}
|
|
)
|
|
|
|
if(WITH_GPU)
|
|
list(APPEND EXTERNAL_LIB ${CUDA_LIBRARIES} ${CUDA_rt_LIBRARY})
|
|
if(NOT WITH_DSO)
|
|
list(APPEND EXTERNAL_LIB ${CUDNN_LIBRARY} ${CUDA_CUBLAS_LIBRARIES} ${CUDA_curand_LIBRARY})
|
|
endif(NOT WITH_DSO)
|
|
endif(WITH_GPU)
|
|
|
|
if(WITH_MKLDNN)
|
|
list(APPEND EXTERNAL_LIBS ${MKLDNN_LIBRARY} ${MKLDNN_IOMP_LIB})
|
|
endif()
|
|
|
|
if(USE_NNPACK)
|
|
include(external/nnpack)
|
|
list(APPEND EXTERNAL_LIBS ${NNPACK_LIBS})
|
|
endif(USE_NNPACK)
|
|
|
|
add_subdirectory(proto)
|
|
|
|
# "add_subdirectory(go)" should be placed after the following loine,
|
|
# because it depends on paddle/optimizer.
|
|
add_subdirectory(paddle/optimizer)
|
|
|
|
# "add_subdirectory(paddle)" and "add_subdirectory(python)" should be
|
|
# placed after this block, because they depends on it.
|
|
if(WITH_GOLANG)
|
|
add_subdirectory(go)
|
|
endif(WITH_GOLANG)
|
|
|
|
add_subdirectory(paddle)
|
|
if(WITH_PYTHON)
|
|
add_subdirectory(python)
|
|
endif()
|
|
if(WITH_DOC)
|
|
add_subdirectory(doc)
|
|
endif()
|