CMakeLists.txt 4.7 KB
Newer Older
1
set(PYBIND_DEPS pybind python proto_desc memory executor fleet_wrapper box_wrapper prune
J
Jiabin Yang 已提交
2
  feed_fetch_method pass_builder parallel_executor profiler layer tracer engine scope_pool
3 4
  analysis_predictor imperative_profiler imperative_flag save_load_util dlpack_tensor device_context
  gloo_wrapper)
5

6 7 8 9
if (WITH_NCCL)
  set(PYBIND_DEPS ${PYBIND_DEPS} nccl_wrapper)
endif()

10
if(NOT WIN32)
11
  set(PYBIND_DEPS ${PYBIND_DEPS} data_loader)
12
  set(PYBIND_DEPS ${PYBIND_DEPS} mmap_allocator)
13 14 15
  if (WITH_NCCL)
    set(PYBIND_DEPS ${PYBIND_DEPS} nccl_context)
  endif()
16
endif(NOT WIN32)
F
flame 已提交
17

S
sneaxiy 已提交
18 19 20
if(WITH_PYTHON)
  list(APPEND PYBIND_DEPS py_func_op)
endif()
21 22 23 24 25 26 27 28 29 30

if (WITH_DISTRIBUTE)
  list(APPEND PYBIND_DEPS communicator)
endif()

set(PYBIND_SRCS
  pybind.cc
  exception.cc
  protobuf.cc
  const_value.cc
31
  global_value_getter_setter.cc
32 33
  reader_py.cc
  fleet_wrapper_py.cc
34
  gloo_wrapper_py.cc
H
hutuxian 已提交
35
  box_helper_py.cc
36 37 38 39 40 41 42 43
  data_set_py.cc
  imperative.cc
  ir.cc
  inference_api.cc)

if (WITH_DISTRIBUTE)
  list(APPEND PYBIND_SRCS communicator_py.cc)
endif()
44

45 46 47 48
if (WITH_NCCL)
  list(APPEND PYBIND_SRCS nccl_wrapper_py.cc)
endif()

L
Leo Chen 已提交
49 50 51 52 53 54
if(WITH_PYTHON)
  # generate op pybind functions automatically for dygraph.
  set(OP_FUNCTION_GENERETOR_DEPS pybind proto_desc executor layer tracer engine imperative_profiler imperative_flag)
  list(APPEND OP_FUNCTION_GENERETOR_DEPS ${GLOB_OP_LIB})
  list(APPEND OP_FUNCTION_GENERETOR_DEPS ${GLOB_OPERATOR_DEPS})

55
  if(WITH_NCCL)
L
Leo Chen 已提交
56
    list(APPEND OP_FUNCTION_GENERETOR_DEPS nccl_context)
57
  endif(WITH_NCCL)
L
Leo Chen 已提交
58 59

  add_executable(op_function_generator op_function_generator.cc)
60
  target_link_libraries(op_function_generator ${OP_FUNCTION_GENERETOR_DEPS})
L
Leo Chen 已提交
61 62 63
  get_property (os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
  target_link_libraries(op_function_generator ${os_dependency_modules})

64
  if(WIN32)
L
Leo Chen 已提交
65 66
    add_custom_target(op_function_cmd 
      COMMAND "${CMAKE_BINARY_DIR}/paddle/fluid/pybind/${CMAKE_BUILD_TYPE}/op_function_generator" 
67 68 69 70 71 72 73
              "${CMAKE_SOURCE_DIR}/paddle/fluid/pybind/op_function_impl.h") 
    add_dependencies(op_function_cmd op_function_generator)
    if(${CBLAS_PROVIDER} STREQUAL MKLML)
      add_custom_command(TARGET op_function_generator POST_BUILD
            COMMAND ${CMAKE_COMMAND} -E copy ${MKLML_SHARED_LIB} ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_BUILD_TYPE}
            COMMAND ${CMAKE_COMMAND} -E copy ${MKLML_SHARED_LIB_DEPS} ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_BUILD_TYPE}
            COMMAND ${CMAKE_COMMAND} -E copy ${MKLML_SHARED_IOMP_LIB} ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_BUILD_TYPE}
L
Leo Chen 已提交
74
          )
75 76 77 78 79
    else(${CBLAS_PROVIDER} STREQUAL EXTERN_OPENBLAS)
      add_custom_command(TARGET op_function_generator POST_BUILD 
            COMMAND ${CMAKE_COMMAND} -E copy ${OPENBLAS_SHARED_LIB} ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_BUILD_TYPE}
          )
    endif()
L
Leo Chen 已提交
80
    if(WITH_MKLDNN)
81
      add_custom_command(TARGET op_function_generator POST_BUILD 
L
Leo Chen 已提交
82 83
          COMMAND ${CMAKE_COMMAND} -E copy ${MKLDNN_SHARED_LIB} ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_BUILD_TYPE}
          )
84
    endif()
L
Leo Chen 已提交
85 86 87 88 89 90 91
  else(WIN32)
    # If there are no *.so in /usr/lib or LD_LIBRARY_PATH, 
    # copy these *.so to current directory and append current directory to
    # LD_LIBRARY_PATH. This is different with Windows platformm, which search 
    # *.dll in current directory automatically.
    add_custom_target(op_function_cmd 
      COMMAND ${CMAKE_COMMAND} -E env "LD_LIBRARY_PATH=$ENV{LD_LIBRARY_PATH}:."
92 93 94
              "${CMAKE_CURRENT_BINARY_DIR}/op_function_generator"
              "${CMAKE_SOURCE_DIR}/paddle/fluid/pybind/op_function_impl.h") 
    add_dependencies(op_function_cmd op_function_generator)
L
Leo Chen 已提交
95
    if(WITH_MKL)
96 97 98 99
      add_custom_command(TARGET op_function_generator POST_BUILD 
            COMMAND ${CMAKE_COMMAND} -E copy ${MKLML_SHARED_LIB} ${CMAKE_CURRENT_BINARY_DIR}
            COMMAND ${CMAKE_COMMAND} -E copy ${MKLML_SHARED_IOMP_LIB} ${CMAKE_CURRENT_BINARY_DIR}
            )
L
Leo Chen 已提交
100 101
    endif(WITH_MKL)
    if(WITH_MKLDNN)
102
      add_custom_command(TARGET op_function_generator POST_BUILD 
L
Leo Chen 已提交
103 104 105 106 107
          COMMAND ${CMAKE_COMMAND} -E copy ${MKLDNN_SHARED_LIB} ${CMAKE_CURRENT_BINARY_DIR}
          )
    endif(WITH_MKLDNN)
  endif(WIN32)

108 109
  if(WITH_AMD_GPU)
    hip_library(paddle_pybind SHARED
D
dzhwinter 已提交
110
      SRCS ${PYBIND_SRCS}
S
sabreshao 已提交
111 112
      DEPS ARCHIVE_START ${PYBIND_DEPS}
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS} ARCHIVE_END)
113 114
  else()
    cc_library(paddle_pybind SHARED
D
dzhwinter 已提交
115
      SRCS ${PYBIND_SRCS}
D
dzhwinter 已提交
116
      DEPS ${PYBIND_DEPS}
W
Wu Yi 已提交
117
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
T
Tao Luo 已提交
118
    if(NOT APPLE AND NOT WIN32)
119
      target_link_libraries(paddle_pybind rt)
T
Tao Luo 已提交
120
    endif(NOT APPLE AND NOT WIN32)
121
  endif(WITH_AMD_GPU)
122

P
peizhilin 已提交
123 124
  get_property (os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
  target_link_libraries(paddle_pybind ${os_dependency_modules})
125
  add_dependencies(paddle_pybind op_function_cmd)
126
endif(WITH_PYTHON)