inference_lib.cmake 6.4 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
# 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.

15 16 17 18
set_property(GLOBAL PROPERTY FLUID_MODULES "")
# find all fluid modules is used for paddle fluid static library
function(find_fluid_modules TARGET_NAME)
  get_filename_component(__target_path ${TARGET_NAME} ABSOLUTE)
19
  string(REGEX REPLACE "^${PADDLE_SOURCE_DIR}/" "" __target_path ${__target_path})
20 21 22 23 24 25 26 27
  string(FIND "${__target_path}" "fluid" pos)
  if(pos GREATER 1)
    get_property(fluid_modules GLOBAL PROPERTY FLUID_MODULES)
    set(fluid_modules ${fluid_modules} ${TARGET_NAME})
    set_property(GLOBAL PROPERTY FLUID_MODULES "${fluid_modules}")
  endif()
endfunction(find_fluid_modules)

L
Luo Tao 已提交
28
# make package for paddle fluid shared and static library
L
Luo Tao 已提交
29 30 31 32 33
function(copy TARGET)
    set(options "")
    set(oneValueArgs "")
    set(multiValueArgs SRCS DSTS DEPS)
    cmake_parse_arguments(copy_lib "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
34
    set(inference_lib_dist_dep ${TARGET} ${inference_lib_dist_dep} PARENT_SCOPE)
L
Luo Tao 已提交
35 36 37 38 39 40 41 42 43 44 45 46

    list(LENGTH copy_lib_SRCS copy_lib_SRCS_len)
    list(LENGTH copy_lib_DSTS copy_lib_DSTS_len)
    if(NOT ${copy_lib_SRCS_len} EQUAL ${copy_lib_DSTS_len})
        message(FATAL_ERROR "${TARGET} source numbers are not equal to destination numbers")
    endif()
    math(EXPR len "${copy_lib_SRCS_len} - 1")
    
    add_custom_target(${TARGET} DEPENDS ${copy_lib_DEPS})
    foreach(index RANGE ${len})
        list(GET copy_lib_SRCS ${index} src)
        list(GET copy_lib_DSTS ${index} dst)
47 48 49 50
        add_custom_command(TARGET ${TARGET} PRE_BUILD 
          COMMAND mkdir -p "${dst}"
          COMMAND cp -r "${src}" "${dst}"
          COMMENT "copying ${src} -> ${dst}")
L
Luo Tao 已提交
51 52 53
    endforeach()
endfunction()

L
Luo Tao 已提交
54
# third party
55
set(dst_dir "${FLUID_INSTALL_DIR}/third_party/eigen3")
L
Luo Tao 已提交
56 57 58
copy(eigen3_lib
  SRCS ${EIGEN_INCLUDE_DIR}/Eigen/Core ${EIGEN_INCLUDE_DIR}/Eigen/src ${EIGEN_INCLUDE_DIR}/unsupported/Eigen
  DSTS ${dst_dir}/Eigen ${dst_dir}/Eigen ${dst_dir}/unsupported
T
tensor-tang 已提交
59
  DEPS eigen3
L
Luo Tao 已提交
60 61
)

62
set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/gflags")
L
Luo Tao 已提交
63 64 65
copy(gflags_lib
  SRCS ${GFLAGS_INCLUDE_DIR} ${GFLAGS_LIBRARIES}
  DSTS ${dst_dir} ${dst_dir}/lib
T
tensor-tang 已提交
66
  DEPS gflags
L
Luo Tao 已提交
67 68
)

69
set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/glog")
L
Luo Tao 已提交
70 71 72
copy(glog_lib
  SRCS ${GLOG_INCLUDE_DIR} ${GLOG_LIBRARIES}
  DSTS ${dst_dir} ${dst_dir}/lib
T
tensor-tang 已提交
73
  DEPS glog
L
Luo Tao 已提交
74 75
)

76
set(dst_dir "${FLUID_INSTALL_DIR}/third_party/boost/")
Q
qiaolongfei 已提交
77
copy(boost_lib
Q
qiaolongfei 已提交
78
  SRCS ${BOOST_INCLUDE_DIR}/boost
Q
qiaolongfei 已提交
79
  DSTS ${dst_dir}
T
tensor-tang 已提交
80
  DEPS boost
Q
qiaolongfei 已提交
81 82
)

83
if(NOT PROTOBUF_FOUND)
84
    set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/protobuf")
L
Luo Tao 已提交
85
    copy(protobuf_lib
86
      SRCS ${PROTOBUF_INCLUDE_DIR} ${PROTOBUF_LIBRARY}
L
Luo Tao 已提交
87
      DSTS ${dst_dir} ${dst_dir}/lib
T
tensor-tang 已提交
88
      DEPS extern_protobuf
L
Luo Tao 已提交
89
    )
90 91 92
endif()

if(NOT CBLAS_FOUND)
93
    set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/openblas")
94 95 96
    copy(openblas_lib
      SRCS ${CBLAS_INSTALL_DIR}/lib ${CBLAS_INSTALL_DIR}/include
      DSTS ${dst_dir} ${dst_dir}
T
tensor-tang 已提交
97
      DEPS extern_openblas
98
    )
L
Luo Tao 已提交
99
elseif (WITH_MKLML)
100
    set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/mklml")
101
    copy(mklml_lib
L
Luo Tao 已提交
102 103
      SRCS ${MKLML_LIB} ${MKLML_IOMP_LIB} ${MKLML_INC_DIR}
      DSTS ${dst_dir}/lib ${dst_dir}/lib ${dst_dir}
T
tensor-tang 已提交
104
      DEPS mklml
105
    )
106
endif()
L
Luo Tao 已提交
107

Q
qiaolongfei 已提交
108
if(WITH_MKLDNN)
109
  set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/mkldnn")
Q
qiaolongfei 已提交
110
  copy(mkldnn_lib
Q
qiaolongfei 已提交
111
    SRCS ${MKLDNN_INC_DIR} ${MKLDNN_SHARED_LIB}
Q
qiaolongfei 已提交
112
    DSTS ${dst_dir} ${dst_dir}/lib
T
tensor-tang 已提交
113
    DEPS mkldnn
Q
qiaolongfei 已提交
114 115 116
  )
endif()

117
if(NOT MOBILE_INFERENCE AND NOT RPI)
118
  set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/snappy")
119 120
  copy(snappy_lib
    SRCS ${SNAPPY_INCLUDE_DIR} ${SNAPPY_LIBRARIES}
T
tensor-tang 已提交
121 122
    DSTS ${dst_dir} ${dst_dir}/lib
    DEPS snappy)
123

124
  set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/snappystream")
125 126
  copy(snappystream_lib
    SRCS ${SNAPPYSTREAM_INCLUDE_DIR} ${SNAPPYSTREAM_LIBRARIES}
T
tensor-tang 已提交
127 128
    DSTS ${dst_dir} ${dst_dir}/lib
    DEPS snappystream)
129

130
  set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/zlib")
131 132
  copy(zlib_lib
    SRCS ${ZLIB_INCLUDE_DIR} ${ZLIB_LIBRARIES}
T
tensor-tang 已提交
133 134
    DSTS ${dst_dir} ${dst_dir}/lib
    DEPS zlib)
135 136
endif()

L
Luo Tao 已提交
137
# paddle fluid module
138
set(src_dir "${PADDLE_SOURCE_DIR}/paddle/fluid")
139
set(dst_dir "${FLUID_INSTALL_DIR}/paddle/fluid")
L
Luo Tao 已提交
140 141
set(module "framework")
copy(framework_lib DEPS framework_py_proto 
142
  SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/details/*.h ${PADDLE_BINARY_DIR}/paddle/fluid/framework/framework.pb.h
L
Luo Tao 已提交
143
  DSTS ${dst_dir}/${module} ${dst_dir}/${module}/details ${dst_dir}/${module}
L
Luo Tao 已提交
144 145
)

L
Luo Tao 已提交
146 147 148 149
set(module "memory")
copy(memory_lib
  SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/detail/*.h
  DSTS ${dst_dir}/${module} ${dst_dir}/${module}/detail
L
Luo Tao 已提交
150 151
)

L
Luo Tao 已提交
152
set(module "inference")
153 154
copy(inference_lib DEPS paddle_fluid_shared paddle_fluid
  SRCS ${src_dir}/${module}/*.h ${PADDLE_BINARY_DIR}/paddle/fluid/inference/libpaddle_fluid.*
L
Luo Tao 已提交
155
  DSTS ${dst_dir}/${module} ${dst_dir}/${module}
L
Luo Tao 已提交
156 157
)

L
Luo Tao 已提交
158
set(module "platform")
X
Xin Pan 已提交
159
copy(platform_lib DEPS profiler_py_proto
L
Luo Tao 已提交
160 161 162
  SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/dynload/*.h ${src_dir}/${module}/details/*.h
  DSTS ${dst_dir}/${module} ${dst_dir}/${module}/dynload ${dst_dir}/${module}/details
)
L
Luo Tao 已提交
163

L
Luo Tao 已提交
164 165 166 167
set(module "string")
copy(string_lib
  SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/tinyformat/*.h
  DSTS ${dst_dir}/${module} ${dst_dir}/${module}/tinyformat
L
Luo Tao 已提交
168 169
)

170 171
set(module "pybind")
copy(pybind_lib
Q
qiaolongfei 已提交
172
  SRCS ${CMAKE_CURRENT_BINARY_DIR}/paddle/fluid/${module}/pybind.h
173 174 175
  DSTS ${dst_dir}/${module}
)

176 177 178
# CMakeCache Info
copy(cmake_cache
  SRCS ${CMAKE_CURRENT_BINARY_DIR}/CMakeCache.txt
179
  DSTS ${FLUID_INSTALL_DIR})
180

181
add_custom_target(inference_lib_dist DEPENDS ${inference_lib_dist_dep}) 
182 183 184 185

# paddle fluid version
execute_process(
  COMMAND ${GIT_EXECUTABLE} log --pretty=format:%H -1
L
Luo Tao 已提交
186
  WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}
187
  OUTPUT_VARIABLE PADDLE_GIT_COMMIT)
188
set(version_file ${FLUID_INSTALL_DIR}/version.txt)
189 190 191 192 193 194 195 196 197
file(WRITE ${version_file}
  "GIT COMMIT ID: ${PADDLE_GIT_COMMIT}\n"
  "WITH_MKL: ${WITH_MKL}\n"
  "WITH_GPU: ${WITH_GPU}\n")
if(WITH_GPU)
  file(APPEND ${version_file}
    "CUDA version: ${CUDA_VERSION}\n"
    "CUDNN version: v${CUDNN_MAJOR_VERSION}\n")
endif()