inference_lib.cmake 3.1 KB
Newer Older
L
Luo Tao 已提交
1
# make package for paddle fluid shared and static library
L
Luo Tao 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
function(copy TARGET)
    set(options "")
    set(oneValueArgs "")
    set(multiValueArgs SRCS DSTS DEPS)
    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)
    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)
19 20 21 22
        add_custom_command(TARGET ${TARGET} PRE_BUILD 
          COMMAND mkdir -p "${dst}"
          COMMAND cp -r "${src}" "${dst}"
          COMMENT "copying ${src} -> ${dst}")
L
Luo Tao 已提交
23 24 25
    endforeach()
endfunction()

L
Luo Tao 已提交
26
# third party
L
Luo Tao 已提交
27 28 29 30
set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/eigen3")
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
L
Luo Tao 已提交
31 32
)

L
Luo Tao 已提交
33 34 35 36
set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/gflags")
copy(gflags_lib
  SRCS ${GFLAGS_INCLUDE_DIR} ${GFLAGS_LIBRARIES}
  DSTS ${dst_dir} ${dst_dir}/lib
L
Luo Tao 已提交
37 38
)

L
Luo Tao 已提交
39 40 41 42
set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/glog")
copy(glog_lib
  SRCS ${GLOG_INCLUDE_DIR} ${GLOG_LIBRARIES}
  DSTS ${dst_dir} ${dst_dir}/lib
L
Luo Tao 已提交
43 44 45
)

IF(NOT PROTOBUF_FOUND)
L
Luo Tao 已提交
46 47 48 49
    set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/protobuf")
    copy(protobuf_lib
      SRCS ${PROTOBUF_INCLUDE_DIR} ${PROTOBUF_LITE_LIBRARY}
      DSTS ${dst_dir} ${dst_dir}/lib
L
Luo Tao 已提交
50 51 52 53
    )
ENDIF(NOT PROTOBUF_FOUND)

# paddle fluid module
54 55
set(src_dir "${PADDLE_SOURCE_DIR}/paddle/fluid")
set(dst_dir "${CMAKE_INSTALL_PREFIX}/paddle/fluid")
L
Luo Tao 已提交
56 57
set(module "framework")
copy(framework_lib DEPS framework_py_proto 
58
  SRCS ${src_dir}/${module}/*.h ${src_dir}/${module}/details/*.h ${PADDLE_BINARY_DIR}/paddle/fluid/framework/framework.pb.h
L
Luo Tao 已提交
59
  DSTS ${dst_dir}/${module} ${dst_dir}/${module}/details ${dst_dir}/${module}
L
Luo Tao 已提交
60 61
)

L
Luo Tao 已提交
62 63 64 65
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 已提交
66 67
)

L
Luo Tao 已提交
68 69
set(module "inference")
copy(inference_lib DEPENDS paddle_fluid_shared
70
  SRCS ${src_dir}/${module}/*.h ${PADDLE_BINARY_DIR}/paddle/fluid/inference/libpaddle_fluid.so
L
Luo Tao 已提交
71
  DSTS ${dst_dir}/${module} ${dst_dir}/${module}
L
Luo Tao 已提交
72 73
)

L
Luo Tao 已提交
74
set(module "platform")
X
Xin Pan 已提交
75
copy(platform_lib DEPS profiler_py_proto
L
Luo Tao 已提交
76 77 78
  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 已提交
79

L
Luo Tao 已提交
80 81 82 83
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 已提交
84 85 86 87 88
)

add_custom_target(inference_lib_dist DEPENDS 
  inference_lib framework_lib memory_lib platform_lib string_lib
  gflags_lib glog_lib protobuf_lib eigen3_lib)