inference_lib.cmake 18.6 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
16 17
set(PADDLE_INSTALL_DIR "${CMAKE_BINARY_DIR}/paddle_install_dir" CACHE STRING
  "A path setting paddle shared and static libraries")
18

19 20
set(PADDLE_INFERENCE_INSTALL_DIR "${CMAKE_BINARY_DIR}/paddle_inference_install_dir" CACHE STRING
  "A path setting paddle inference shared and static libraries")
21
  
22 23
# At present, the size of static lib in Windows is very large,
# so we need to crop the library size.
24
if(WIN32)
25
    #todo: remove the option 
26
    option(WITH_STATIC_LIB "Compile demo with static/shared library, default use dynamic."   OFF)
27
    if(NOT PYTHON_EXECUTABLE)
28
        FIND_PACKAGE(PythonInterp REQUIRED)
29 30 31
    endif()
endif()

32
set(COPY_SCRIPT_DIR ${PADDLE_SOURCE_DIR}/cmake)
L
Luo Tao 已提交
33 34 35
function(copy TARGET)
    set(options "")
    set(oneValueArgs "")
36
    set(multiValueArgs SRCS DSTS)
L
Luo Tao 已提交
37 38 39 40
    cmake_parse_arguments(copy_lib "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})

    list(LENGTH copy_lib_SRCS copy_lib_SRCS_len)
    list(LENGTH copy_lib_DSTS copy_lib_DSTS_len)
41
    if (NOT ${copy_lib_SRCS_len} EQUAL ${copy_lib_DSTS_len})
L
Luo Tao 已提交
42
        message(FATAL_ERROR "${TARGET} source numbers are not equal to destination numbers")
43
    endif ()
L
Luo Tao 已提交
44
    math(EXPR len "${copy_lib_SRCS_len} - 1")
45
    foreach (index RANGE ${len})
L
Luo Tao 已提交
46 47
        list(GET copy_lib_SRCS ${index} src)
        list(GET copy_lib_DSTS ${index} dst)
48 49 50 51 52 53
        if (WIN32)   #windows
            file(TO_NATIVE_PATH ${src} native_src)
            file(TO_NATIVE_PATH ${dst} native_dst)
            add_custom_command(TARGET ${TARGET} POST_BUILD
                    COMMAND ${PYTHON_EXECUTABLE} ${COPY_SCRIPT_DIR}/copyfile.py ${native_src} ${native_dst})
        else (WIN32) #not windows
54
            add_custom_command(TARGET ${TARGET} POST_BUILD
55 56 57 58 59
                    COMMAND mkdir -p "${dst}"
                    COMMAND cp -r "${src}" "${dst}"
                    COMMENT "copying ${src} -> ${dst}")
        endif (WIN32) # not windows
    endforeach ()
L
Luo Tao 已提交
60 61
endfunction()

62 63 64 65 66 67
function(copy_part_of_thrid_party TARGET DST) 
    if(${CBLAS_PROVIDER} STREQUAL MKLML)
        set(dst_dir "${DST}/third_party/install/mklml")
        if(WIN32)
            copy(${TARGET}
                    SRCS ${MKLML_LIB} ${MKLML_IOMP_LIB} ${MKLML_SHARED_LIB}
68
                    ${MKLML_SHARED_IOMP_LIB} ${MKLML_INC_DIR}
69
                    DSTS ${dst_dir}/lib ${dst_dir}/lib ${dst_dir}/lib
70
                    ${dst_dir}/lib ${dst_dir})
71 72 73 74
        else()
            copy(${TARGET}
                    SRCS ${MKLML_LIB} ${MKLML_IOMP_LIB} ${MKLML_INC_DIR}
                    DSTS ${dst_dir}/lib ${dst_dir}/lib ${dst_dir})
75 76 77 78 79 80
            if(WITH_STRIP)
                    add_custom_command(TARGET ${TARGET} POST_BUILD
                            COMMAND strip -s ${dst_dir}/lib/libiomp5.so
                            COMMAND strip -s ${dst_dir}/lib/libmklml_intel.so
                            COMMENT "striping libiomp5.so\nstriping libmklml_intel.so")
            endif()
81 82 83
        endif()
    elseif(${CBLAS_PROVIDER} STREQUAL EXTERN_OPENBLAS)
        set(dst_dir "${DST}/third_party/install/openblas")
84 85 86 87 88
	if(WIN32)
            copy(${TARGET}
                    SRCS ${CBLAS_INSTALL_DIR}/lib ${OPENBLAS_SHARED_LIB} ${CBLAS_INSTALL_DIR}/include
                    DSTS ${dst_dir} ${dst_dir}/lib ${dst_dir})
	else()
89 90 91
            copy(${TARGET}
                    SRCS ${CBLAS_INSTALL_DIR}/lib ${CBLAS_INSTALL_DIR}/include
                    DSTS ${dst_dir} ${dst_dir})
92
	endif()
93 94 95 96 97 98
    endif()

    if(WITH_MKLDNN)
        set(dst_dir "${DST}/third_party/install/mkldnn")
        if(WIN32)
            copy(${TARGET}
A
Adam 已提交
99
                    SRCS ${MKLDNN_INC_DIR} ${MKLDNN_SHARED_LIB}  ${MKLDNN_LIB}
100 101 102
                    DSTS ${dst_dir} ${dst_dir}/lib ${dst_dir}/lib)
        else()
            copy(${TARGET}
103 104 105 106 107 108 109 110 111 112 113
                    SRCS ${MKLDNN_INC_DIR} ${MKLDNN_SHARED_LIB}
                    DSTS ${dst_dir} ${dst_dir}/lib)
            if(WITH_STRIP)
                    add_custom_command(TARGET ${TARGET} POST_BUILD
                            COMMAND strip -s ${dst_dir}/lib/libmkldnn.so.0
                            COMMENT "striping libmkldnn.so.0")
            endif()
            add_custom_command(TARGET ${TARGET} POST_BUILD
                    COMMAND ${CMAKE_COMMAND} -E create_symlink libmkldnn.so.0 ${dst_dir}/lib/libdnnl.so.1
                    COMMAND ${CMAKE_COMMAND} -E create_symlink libmkldnn.so.0 ${dst_dir}/lib/libdnnl.so.2
                    COMMENT "Make a symbol link of libmkldnn.so.0")
114 115 116
        endif()
    endif()

117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134
    if (WITH_ONNXRUNTIME)
        set(dst_dir "${DST}/third_party/install/onnxruntime")
        copy(${TARGET}
                SRCS ${ONNXRUNTIME_INC_DIR} ${ONNXRUNTIME_LIB_DIR}
                DSTS ${dst_dir} ${dst_dir})

        set(dst_dir "${DST}/third_party/install/paddle2onnx")
        if(WIN32)
            copy(${TARGET}
                SRCS ${PADDLE2ONNX_INC_DIR}/paddle2onnx ${PADDLE2ONNX_SHARED_LIB} ${PADDLE2ONNX_LIB}
                DSTS ${dst_dir}/include ${dst_dir}/lib ${dst_dir}/lib)
        else()
            copy(${TARGET}
                SRCS ${PADDLE2ONNX_INC_DIR}/paddle2onnx ${PADDLE2ONNX_LIB}
                DSTS ${dst_dir}/include ${dst_dir}/lib)
        endif()
    endif()

135 136 137 138 139 140 141 142 143
    set(dst_dir "${DST}/third_party/install/gflags")
    copy(${TARGET}
            SRCS ${GFLAGS_INCLUDE_DIR} ${GFLAGS_LIBRARIES}
            DSTS ${dst_dir} ${dst_dir}/lib)

    set(dst_dir "${DST}/third_party/install/glog")
    copy(${TARGET}
            SRCS ${GLOG_INCLUDE_DIR} ${GLOG_LIBRARIES}
            DSTS ${dst_dir} ${dst_dir}/lib)
144

S
Steffy-zxf 已提交
145 146 147 148 149
    set(dst_dir "${DST}/third_party/install/utf8proc")
    copy(${TARGET}
            SRCS ${UTF8PROC_INSTALL_DIR}/include ${UTF8PROC_LIBRARIES}
            DSTS ${dst_dir} ${dst_dir}/lib)

150
    if (WITH_CRYPTO)
M
MRXLT 已提交
151 152
        set(dst_dir "${DST}/third_party/install/cryptopp")
        copy(${TARGET}
153 154 155
            SRCS ${CRYPTOPP_INCLUDE_DIR} ${CRYPTOPP_LIBRARIES}
            DSTS ${dst_dir} ${dst_dir}/lib)
    endif()
M
MRXLT 已提交
156

157 158 159 160
    set(dst_dir "${DST}/third_party/install/xxhash")
    copy(${TARGET}
        SRCS ${XXHASH_INCLUDE_DIR} ${XXHASH_LIBRARIES}
        DSTS ${dst_dir} ${dst_dir}/lib)    
161

162 163 164 165 166 167 168
    if (NOT PROTOBUF_FOUND OR WIN32)
        set(dst_dir "${DST}/third_party/install/protobuf")
        copy(${TARGET}
                SRCS ${PROTOBUF_INCLUDE_DIR} ${PROTOBUF_LIBRARY}
                DSTS ${dst_dir} ${dst_dir}/lib)
    endif ()

169 170 171
    if (LITE_BINARY_DIR)
        set(dst_dir "${DST}/third_party/install/lite")
        copy(${TARGET}
W
Wilber 已提交
172
                SRCS ${LITE_BINARY_DIR}/${LITE_OUTPUT_BIN_DIR}/*
173 174
                DSTS ${dst_dir})
    endif()
175 176
endfunction()

177
# inference library for only inference
178
set(inference_lib_deps third_party paddle_inference paddle_inference_c paddle_inference_shared paddle_inference_c_shared)
179
add_custom_target(inference_lib_dist DEPENDS ${inference_lib_deps})
180

181

182
set(dst_dir "${PADDLE_INFERENCE_INSTALL_DIR}/third_party/threadpool")
183 184 185 186
copy(inference_lib_dist
        SRCS ${THREADPOOL_INCLUDE_DIR}/ThreadPool.h
        DSTS ${dst_dir})

187
# GPU must copy externalErrorMsg.pb
188
IF(WITH_GPU)
189 190 191 192
    set(dst_dir "${PADDLE_INFERENCE_INSTALL_DIR}/third_party/externalError/data")
    copy(inference_lib_dist
            SRCS ${externalError_INCLUDE_DIR}
            DSTS ${dst_dir})
193
ENDIF()
194

195 196 197 198 199 200 201
IF(WITH_XPU)
    set(dst_dir "${PADDLE_INFERENCE_INSTALL_DIR}/third_party/install/xpu")
    copy(inference_lib_dist
        SRCS ${XPU_INC_DIR} ${XPU_LIB_DIR}
        DSTS ${dst_dir} ${dst_dir})
ENDIF()

202
# CMakeCache Info
203
copy(inference_lib_dist
204
        SRCS ${CMAKE_CURRENT_BINARY_DIR}/CMakeCache.txt
205
        DSTS ${PADDLE_INFERENCE_INSTALL_DIR})
206

207
copy_part_of_thrid_party(inference_lib_dist ${PADDLE_INFERENCE_INSTALL_DIR})
208

209
set(src_dir "${PADDLE_SOURCE_DIR}/paddle/fluid")
210

P
peizhilin 已提交
211
if(WIN32)
212
    if(WITH_STATIC_LIB)
213 214
        set(paddle_inference_lib $<TARGET_FILE_DIR:paddle_inference>/libpaddle_inference.lib
                             $<TARGET_FILE_DIR:paddle_inference>/paddle_inference.*)
215
    else()
216 217
        set(paddle_inference_lib $<TARGET_FILE_DIR:paddle_inference_shared>/paddle_inference.dll
                             $<TARGET_FILE_DIR:paddle_inference_shared>/paddle_inference.lib)
218
    endif()
219
    copy(inference_lib_dist
220
            SRCS  ${src_dir}/inference/api/paddle_*.h ${paddle_inference_lib}
221 222
            DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/lib
            ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/lib)
P
peizhilin 已提交
223
else(WIN32)
224
    set(paddle_inference_lib ${PADDLE_BINARY_DIR}/paddle/fluid/inference/libpaddle_inference.*)
225
    copy(inference_lib_dist
226
                SRCS  ${src_dir}/inference/api/paddle_*.h ${paddle_inference_lib}
227
                DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/lib)
228
endif(WIN32)
229

230 231
copy(inference_lib_dist
        SRCS  ${CMAKE_BINARY_DIR}/paddle/fluid/framework/framework.pb.h
232
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/internal)
M
MRXLT 已提交
233
copy(inference_lib_dist
M
MRXLT 已提交
234
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/fluid/framework/io/crypto/cipher.h
235
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/crypto/)
M
MRXLT 已提交
236
include_directories(${CMAKE_BINARY_DIR}/../paddle/fluid/framework/io)
237

238
# copy api headers for phi & custom op
239
copy(inference_lib_dist
240 241
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/phi/api/ext/*.h
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/experimental/phi/api/ext/)
242
copy(inference_lib_dist
243 244
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/phi/api/include/*.h
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/experimental/phi/api/include/)
245
copy(inference_lib_dist
246 247
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/phi/api/all.h
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/experimental/phi/api/)
248
copy(inference_lib_dist
249 250
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/phi/common/*.h
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/experimental/phi/common/)
251 252 253
copy(inference_lib_dist
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/phi/core/macros.h
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/experimental/phi/core/)
254 255
copy(inference_lib_dist
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/utils/any.h
256
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/experimental/utils/)
257 258 259
copy(inference_lib_dist
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/utils/optional.h
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/experimental/utils/)
260
copy(inference_lib_dist
261 262
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/utils/none.h
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/experimental/utils/)
263 264 265
copy(inference_lib_dist
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/utils/flat_hash_map.h
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/experimental/utils/)
266 267
copy(inference_lib_dist
        SRCS  ${PADDLE_SOURCE_DIR}/paddle/extension.h
268
        DSTS  ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/experimental/)
269

270 271
# the header file of phi is copied to the experimental directory,
# the include path of phi needs to be changed to adapt to inference api path
272
add_custom_command(TARGET inference_lib_dist POST_BUILD
273 274
        COMMAND ${CMAKE_COMMAND} -P "${PADDLE_SOURCE_DIR}/cmake/phi_header.cmake"
        COMMENT "Change phi header include path to adapt to inference api path")
275

276
# CAPI inference library for only inference
277 278 279
set(PADDLE_INFERENCE_C_INSTALL_DIR "${CMAKE_BINARY_DIR}/paddle_inference_c_install_dir" CACHE STRING
"A path setting CAPI paddle inference shared")
copy_part_of_thrid_party(inference_lib_dist ${PADDLE_INFERENCE_C_INSTALL_DIR})
280 281

set(src_dir "${PADDLE_SOURCE_DIR}/paddle/fluid")
282
if(WIN32)
283
  set(paddle_inference_c_lib $<TARGET_FILE_DIR:paddle_inference_c>/paddle_inference_c.*)
284
else(WIN32)
285
  set(paddle_inference_c_lib ${PADDLE_BINARY_DIR}/paddle/fluid/inference/capi_exp/libpaddle_inference_c.*)
286
endif(WIN32)
287

288
copy(inference_lib_dist
289
      SRCS  ${src_dir}/inference/capi_exp/pd_*.h  ${paddle_inference_c_lib}
290
      DSTS  ${PADDLE_INFERENCE_C_INSTALL_DIR}/paddle/include ${PADDLE_INFERENCE_C_INSTALL_DIR}/paddle/lib)
291

292 293 294 295 296 297 298
if(WITH_STRIP AND NOT WIN32)
        add_custom_command(TARGET inference_lib_dist POST_BUILD
                COMMAND strip -s ${PADDLE_INFERENCE_C_INSTALL_DIR}/paddle/lib/libpaddle_inference_c.so
                COMMAND strip -s ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/lib/libpaddle_inference.so
                COMMENT "striping libpaddle_inference_c.so\nstriping libpaddle_inference.so")
endif()

299 300 301 302
# fluid library for both train and inference
set(fluid_lib_deps inference_lib_dist)
add_custom_target(fluid_lib_dist ALL DEPENDS ${fluid_lib_deps})

303
set(dst_dir "${PADDLE_INSTALL_DIR}/paddle/fluid")
304
set(module "inference")
305
if(WIN32)
306
        copy(fluid_lib_dist
307
                SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/api/paddle_*.h ${paddle_inference_lib}
308 309
                DSTS ${dst_dir}/${module} ${dst_dir}/${module} ${dst_dir}/${module} ${dst_dir}/${module}
                )
310
        else()
311
        copy(fluid_lib_dist
312
                SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/api/paddle_*.h ${paddle_inference_lib}
313 314 315
                DSTS ${dst_dir}/${module} ${dst_dir}/${module} ${dst_dir}/${module} 
                )
endif()
316 317

set(module "framework")
318
set(framework_lib_deps framework_proto data_feed_proto trainer_desc_proto)
319 320
add_dependencies(fluid_lib_dist ${framework_lib_deps})
copy(fluid_lib_dist
321 322 323 324 325 326 327 328 329
        SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/details/*.h ${PADDLE_BINARY_DIR}/paddle/fluid/framework/trainer_desc.pb.h ${PADDLE_BINARY_DIR}/paddle/fluid/framework/framework.pb.h ${PADDLE_BINARY_DIR}/paddle/fluid/framework/data_feed.pb.h ${src_dir}/${module}/ir/memory_optimize_pass/*.h
        ${src_dir}/${module}/ir/*.h ${src_dir}/${module}/fleet/*.h
        DSTS ${dst_dir}/${module} ${dst_dir}/${module}/details ${dst_dir}/${module} ${dst_dir}/${module} ${dst_dir}/${module} ${dst_dir}/${module}/ir/memory_optimize_pass ${dst_dir}/${module}/ir ${dst_dir}/${module}/fleet)

set(module "operators")
copy(fluid_lib_dist
        SRCS ${src_dir}/${module}/reader/blocking_queue.h
        DSTS ${dst_dir}/${module}/reader/
        )
330 331 332 333 334

set(module "memory")
copy(fluid_lib_dist
        SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/detail/*.h ${src_dir}/${module}/allocation/*.h
        DSTS ${dst_dir}/${module} ${dst_dir}/${module}/detail ${dst_dir}/${module}/allocation
P
peizhilin 已提交
335
        )
Y
Yan Chunwei 已提交
336

L
Luo Tao 已提交
337
set(module "platform")
338
set(platform_lib_deps profiler_proto errors)
339
if(WITH_GPU)
340
  set(platform_lib_deps ${platform_lib_deps} external_error_proto)
341 342
endif(WITH_GPU)

343 344
add_dependencies(fluid_lib_dist ${platform_lib_deps})
copy(fluid_lib_dist
345 346
        SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/dynload/*.h ${src_dir}/${module}/details/*.h ${PADDLE_BINARY_DIR}/paddle/fluid/platform/*.pb.h
        DSTS ${dst_dir}/${module} ${dst_dir}/${module}/dynload ${dst_dir}/${module}/details ${dst_dir}/${module}
347
        )
L
Luo Tao 已提交
348

L
Luo Tao 已提交
349
set(module "string")
350
copy(fluid_lib_dist
351
        SRCS ${PADDLE_SOURCE_DIR}/paddle/utils/${module}/*.h ${PADDLE_SOURCE_DIR}/paddle/utils/${module}/tinyformat/*.h 
352 353
        DSTS ${dst_dir}/${module} ${dst_dir}/${module}/tinyformat
        )
L
Luo Tao 已提交
354

355 356 357 358 359 360
set(module "imperative")
copy(fluid_lib_dist
        SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/jit/*.h 
        DSTS ${dst_dir}/${module} ${dst_dir}/${module}/jit
        )

361
set(module "pybind")
362
copy(fluid_lib_dist
363 364 365
        SRCS ${CMAKE_CURRENT_BINARY_DIR}/paddle/fluid/${module}/pybind.h
        DSTS ${dst_dir}/${module}
        )
366

367
set(dst_dir "${PADDLE_INSTALL_DIR}/third_party/eigen3")
368 369 370 371
copy(inference_lib_dist
        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)

372
set(dst_dir "${PADDLE_INSTALL_DIR}/third_party/boost")
373 374 375 376
copy(inference_lib_dist
        SRCS ${BOOST_INCLUDE_DIR}/boost
        DSTS ${dst_dir})

377
set(dst_dir "${PADDLE_INSTALL_DIR}/third_party/dlpack")
378 379 380 381
copy(inference_lib_dist
        SRCS ${DLPACK_INCLUDE_DIR}/dlpack
        DSTS ${dst_dir})

382
set(dst_dir "${PADDLE_INSTALL_DIR}/third_party/install/zlib")
383 384 385 386
copy(inference_lib_dist
        SRCS ${ZLIB_INCLUDE_DIR} ${ZLIB_LIBRARIES}
        DSTS ${dst_dir} ${dst_dir}/lib)

387

388
# CMakeCache Info
389
copy(fluid_lib_dist
390 391
        SRCS ${PADDLE_INFERENCE_INSTALL_DIR}/third_party ${CMAKE_CURRENT_BINARY_DIR}/CMakeCache.txt
        DSTS ${PADDLE_INSTALL_DIR} ${PADDLE_INSTALL_DIR}
392
        )
T
Tao Luo 已提交
393

394
# paddle fluid version
T
Tao Luo 已提交
395
function(version version_file)
396 397 398 399 400 401 402 403
    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"
404
            "WITH_GPU: ${WITH_GPU}\n"
405 406
            "WITH_ROCM: ${WITH_ROCM}\n"
            "WITH_ASCEND_CL: ${WITH_ASCEND_CL}\n"
A
Allen Guo 已提交
407 408
            "WITH_ASCEND_CXX11: ${WITH_ASCEND_CXX11}\n"
            "WITH_IPU: ${WITH_IPU}\n")
409
    if(WITH_GPU)
410 411
        file(APPEND ${version_file}
                "CUDA version: ${CUDA_VERSION}\n"
412
                "CUDNN version: v${CUDNN_MAJOR_VERSION}.${CUDNN_MINOR_VERSION}\n")
413
    endif()
414 415 416 417 418
    if(WITH_ROCM)
        file(APPEND ${version_file}
                "HIP version: ${HIP_VERSION}\n"
                "MIOpen version: v${MIOPEN_MAJOR_VERSION}.${MIOPEN_MINOR_VERSION}\n")
    endif()
419 420 421 422 423
    if(WITH_ASCEND_CL)
        file(APPEND ${version_file}
                "Ascend Toolkit version: ${ASCEND_TOOLKIT_VERSION}\n"
                "Ascend Driver version: ${ASCEND_DRIVER_VERSION}\n")
    endif()
A
Allen Guo 已提交
424 425 426 427
    if(WITH_IPU)
        file(APPEND ${version_file}
                "PopART version: ${POPART_VERSION}\n")
    endif()
428
    file(APPEND ${version_file} "CXX compiler version: ${CMAKE_CXX_COMPILER_VERSION}\n")
429 430
    if(TENSORRT_FOUND)
        file(APPEND ${version_file}
431
                "WITH_TENSORRT: ${TENSORRT_FOUND}\n" "TensorRT version: v${TENSORRT_MAJOR_VERSION}.${TENSORRT_MINOR_VERSION}.${TENSORRT_PATCH_VERSION}.${TENSORRT_BUILD_VERSION}\n")
432
    endif()
433
    if(WITH_LITE)
W
Wilber 已提交
434
        file(APPEND ${version_file} "WITH_LITE: ${WITH_LITE}\n" "LITE_GIT_TAG: ${LITE_GIT_TAG}\n")
435
    endif()
436
    
T
Tao Luo 已提交
437
endfunction()
438 439 440
version(${PADDLE_INSTALL_DIR}/version.txt)
version(${PADDLE_INFERENCE_INSTALL_DIR}/version.txt)
version(${PADDLE_INFERENCE_C_INSTALL_DIR}/version.txt)