inference_lib.cmake 7.8 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.

L
Luo Tao 已提交
15
# make package for paddle fluid shared and static library
L
Luo Tao 已提交
16 17 18 19 20
function(copy TARGET)
    set(options "")
    set(oneValueArgs "")
    set(multiValueArgs SRCS DSTS DEPS)
    cmake_parse_arguments(copy_lib "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
21
    set(fluid_lib_dist_dep ${TARGET} ${fluid_lib_dist_dep} PARENT_SCOPE)
L
Luo Tao 已提交
22 23 24 25 26 27 28

    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")
29

L
Luo Tao 已提交
30 31 32 33
    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)
34
        add_custom_command(TARGET ${TARGET} PRE_BUILD
35 36 37
          COMMAND mkdir -p "${dst}"
          COMMAND cp -r "${src}" "${dst}"
          COMMENT "copying ${src} -> ${dst}")
L
Luo Tao 已提交
38 39 40
    endforeach()
endfunction()

L
Luo Tao 已提交
41
# third party
42
set(dst_dir "${FLUID_INSTALL_DIR}/third_party/eigen3")
L
Luo Tao 已提交
43 44 45
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 已提交
46
  DEPS eigen3
L
Luo Tao 已提交
47 48
)

49
set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/gflags")
L
Luo Tao 已提交
50 51 52
copy(gflags_lib
  SRCS ${GFLAGS_INCLUDE_DIR} ${GFLAGS_LIBRARIES}
  DSTS ${dst_dir} ${dst_dir}/lib
T
tensor-tang 已提交
53
  DEPS gflags
L
Luo Tao 已提交
54 55
)

56
set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/glog")
L
Luo Tao 已提交
57 58 59
copy(glog_lib
  SRCS ${GLOG_INCLUDE_DIR} ${GLOG_LIBRARIES}
  DSTS ${dst_dir} ${dst_dir}/lib
T
tensor-tang 已提交
60
  DEPS glog
L
Luo Tao 已提交
61 62
)

63
set(dst_dir "${FLUID_INSTALL_DIR}/third_party/boost/")
Q
qiaolongfei 已提交
64
copy(boost_lib
Q
qiaolongfei 已提交
65
  SRCS ${BOOST_INCLUDE_DIR}/boost
Q
qiaolongfei 已提交
66
  DSTS ${dst_dir}
T
tensor-tang 已提交
67
  DEPS boost
Q
qiaolongfei 已提交
68 69
)

70 71 72 73 74 75 76
set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/xxhash")
copy(xxhash_lib
  SRCS ${XXHASH_INCLUDE_DIR} ${XXHASH_LIBRARIES}
  DSTS ${dst_dir} ${dst_dir}/lib
  DEPS xxhash
)

77
if(NOT PROTOBUF_FOUND)
78
    set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/protobuf")
L
Luo Tao 已提交
79
    copy(protobuf_lib
80
      SRCS ${PROTOBUF_INCLUDE_DIR} ${PROTOBUF_LIBRARY}
L
Luo Tao 已提交
81
      DSTS ${dst_dir} ${dst_dir}/lib
T
tensor-tang 已提交
82
      DEPS extern_protobuf
L
Luo Tao 已提交
83
    )
84 85 86
endif()

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

Q
qiaolongfei 已提交
102
if(WITH_MKLDNN)
103
  set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/mkldnn")
Q
qiaolongfei 已提交
104
  copy(mkldnn_lib
Q
qiaolongfei 已提交
105
    SRCS ${MKLDNN_INC_DIR} ${MKLDNN_SHARED_LIB}
Q
qiaolongfei 已提交
106
    DSTS ${dst_dir} ${dst_dir}/lib
T
tensor-tang 已提交
107
    DEPS mkldnn
Q
qiaolongfei 已提交
108 109 110
  )
endif()

D
dzhwinter 已提交
111
if (NOT WIN32)
112
if(NOT MOBILE_INFERENCE AND NOT RPI)
113
  set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/snappy")
114 115
  copy(snappy_lib
    SRCS ${SNAPPY_INCLUDE_DIR} ${SNAPPY_LIBRARIES}
T
tensor-tang 已提交
116 117
    DSTS ${dst_dir} ${dst_dir}/lib
    DEPS snappy)
118

119
  set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/snappystream")
120 121
  copy(snappystream_lib
    SRCS ${SNAPPYSTREAM_INCLUDE_DIR} ${SNAPPYSTREAM_LIBRARIES}
T
tensor-tang 已提交
122 123
    DSTS ${dst_dir} ${dst_dir}/lib
    DEPS snappystream)
124

125
  set(dst_dir "${FLUID_INSTALL_DIR}/third_party/install/zlib")
126 127
  copy(zlib_lib
    SRCS ${ZLIB_INCLUDE_DIR} ${ZLIB_LIBRARIES}
T
tensor-tang 已提交
128 129
    DSTS ${dst_dir} ${dst_dir}/lib
    DEPS zlib)
130
endif()
D
dzhwinter 已提交
131
endif(NOT WIN32)
132

L
Luo Tao 已提交
133
# paddle fluid module
134
set(src_dir "${PADDLE_SOURCE_DIR}/paddle/fluid")
135
set(dst_dir "${FLUID_INSTALL_DIR}/paddle/fluid")
L
Luo Tao 已提交
136
set(module "framework")
D
dzhwinter 已提交
137
if (NOT WIN32)
L
luotao1 已提交
138 139 140
set(framework_lib_deps framework_py_proto)
endif(NOT WIN32)
copy(framework_lib DEPS ${framework_lib_deps}
D
dzhwinter 已提交
141
  SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/details/*.h ${PADDLE_BINARY_DIR}/paddle/fluid/framework/framework.pb.h
L
luotao1 已提交
142 143
       ${src_dir}/${module}/ir/*.h
  DSTS ${dst_dir}/${module} ${dst_dir}/${module}/details ${dst_dir}/${module} ${dst_dir}/${module}/ir
D
dzhwinter 已提交
144
)
L
Luo Tao 已提交
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
)

Y
Yan Chunwei 已提交
152
set(inference_deps paddle_fluid_shared paddle_fluid)
L
Luo Tao 已提交
153

154
set(module "inference/api")
155
if (WITH_ANAKIN AND WITH_MKL)
156 157 158
    copy(anakin_inference_lib DEPS paddle_inference_api inference_anakin_api
        SRCS
        ${PADDLE_BINARY_DIR}/paddle/fluid/inference/api/libinference_anakin_api* # compiled anakin api
L
Luo Tao 已提交
159
        ${ANAKIN_INSTALL_DIR} # anakin release
T
Tao Luo 已提交
160
        DSTS ${FLUID_INSTALL_DIR}/third_party/install/anakin ${FLUID_INSTALL_DIR}/third_party/install/anakin)
161
     list(APPEND inference_deps anakin_inference_lib)
162 163
endif()

Y
Yan Chunwei 已提交
164 165 166
set(module "inference")
copy(inference_lib DEPS ${inference_deps}
  SRCS ${src_dir}/${module}/*.h ${PADDLE_BINARY_DIR}/paddle/fluid/inference/libpaddle_fluid.*
T
Tao Luo 已提交
167
       ${src_dir}/${module}/api/paddle_inference_api.h
L
luotao1 已提交
168
       ${PADDLE_BINARY_DIR}/paddle/fluid/inference/api/paddle_inference_pass.h
T
Tao Luo 已提交
169
  DSTS ${dst_dir}/${module} ${dst_dir}/${module} ${dst_dir}/${module} ${dst_dir}/${module}
Y
Yan Chunwei 已提交
170 171
)

L
Luo Tao 已提交
172
set(module "platform")
X
Xin Pan 已提交
173
copy(platform_lib DEPS profiler_py_proto
L
Luo Tao 已提交
174 175 176
  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 已提交
177

L
Luo Tao 已提交
178 179 180 181
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 已提交
182 183
)

184 185
set(module "pybind")
copy(pybind_lib
Q
qiaolongfei 已提交
186
  SRCS ${CMAKE_CURRENT_BINARY_DIR}/paddle/fluid/${module}/pybind.h
187 188 189
  DSTS ${dst_dir}/${module}
)

190 191 192
# CMakeCache Info
copy(cmake_cache
  SRCS ${CMAKE_CURRENT_BINARY_DIR}/CMakeCache.txt
193
  DSTS ${FLUID_INSTALL_DIR})
194

195
# This command generates a complete fluid library for both train and inference
196
add_custom_target(fluid_lib_dist DEPENDS ${fluid_lib_dist_dep})
197

T
Tao Luo 已提交
198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213
# Following commands generate a inference-only fluid library
# third_party, version.txt and CMakeCache.txt are the same position with ${FLUID_INSTALL_DIR}
copy(third_party DEPS fluid_lib_dist
  SRCS ${FLUID_INSTALL_DIR}/third_party ${FLUID_INSTALL_DIR}/CMakeCache.txt
  DSTS ${FLUID_INFERENCE_INSTALL_DIR} ${FLUID_INFERENCE_INSTALL_DIR}
)

# only need libpaddle_fluid.so/a and paddle_inference_api.h for inference-only library
copy(inference_api_lib DEPS fluid_lib_dist
  SRCS ${FLUID_INSTALL_DIR}/paddle/fluid/inference/libpaddle_fluid.*
       ${FLUID_INSTALL_DIR}/paddle/fluid/inference/paddle_inference_api.h
  DSTS ${FLUID_INFERENCE_INSTALL_DIR}/paddle/lib ${FLUID_INFERENCE_INSTALL_DIR}/paddle/include
)

add_custom_target(inference_lib_dist DEPENDS third_party inference_api_lib)

214
# paddle fluid version
T
Tao Luo 已提交
215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232
function(version version_file)
  execute_process(
    COMMAND ${GIT_EXECUTABLE} log --pretty=format:%H -1
    WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}
    OUTPUT_VARIABLE PADDLE_GIT_COMMIT)
  file(WRITE ${version_file}
    "GIT COMMIT ID: ${PADDLE_GIT_COMMIT}\n"
    "WITH_MKL: ${WITH_MKL}\n"
    "WITH_MKLDNN: ${WITH_MKLDNN}\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()
endfunction()
version(${FLUID_INSTALL_DIR}/version.txt)
version(${FLUID_INFERENCE_INSTALL_DIR}/version.txt)