CMakeLists.txt 6.2 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
  analysis_predictor imperative_profiler imperative_flag save_load_util dlpack_tensor device_context
T
Thunderbrook 已提交
4
  gloo_wrapper infer_io_utils heter_wrapper generator op_version_registry ps_gpu_wrapper)
5

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

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

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

set(PYBIND_SRCS
  pybind.cc
  exception.cc
  protobuf.cc
  const_value.cc
28
  global_value_getter_setter.cc
29 30
  reader_py.cc
  fleet_wrapper_py.cc
T
Thunderbrook 已提交
31
  heter_wrapper_py.cc
T
Thunderbrook 已提交
32
  ps_gpu_wrapper_py.cc
33
  gloo_wrapper_py.cc
H
hutuxian 已提交
34
  box_helper_py.cc
35 36 37
  data_set_py.cc
  imperative.cc
  ir.cc
Y
yaoxuefeng 已提交
38
  inference_api.cc
39
  compatible.cc
Y
yaoxuefeng 已提交
40
  generator_py.cc)
41

42 43 44 45 46
if(WITH_ASCEND)
  set(PYBIND_DEPS ${PYBIND_DEPS} ascend_wrapper)
  set(PYBIND_SRCS ${PYBIND_SRCS} ascend_wrapper_py.cc)
endif(WITH_ASCEND)

47 48 49 50 51
if(WITH_GLOO)
  set(PYBIND_DEPS ${PYBIND_DEPS} gloo_context)
  set(PYBIND_SRCS ${PYBIND_SRCS} gloo_context_py.cc)
endif(WITH_GLOO)

Y
Yanghello 已提交
52 53 54 55 56
if (WITH_CRYPTO)
  set(PYBIND_DEPS ${PYBIND_DEPS} paddle_crypto)
  set(PYBIND_SRCS ${PYBIND_SRCS} crypto.cc)
endif (WITH_CRYPTO)

T
tangwei12 已提交
57
if (WITH_PSCORE)
T
tangwei12 已提交
58 59 60 61
  set(DISTRIBUTE_COMPILE_FLAGS "-Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor -Wno-error=sign-compare -Wno-error=unused-variable -Wno-error=return-type -Wno-error=unused-but-set-variable -Wno-error=type-limits -Wno-error=unknown-pragmas -Wno-error=parentheses -Wno-error=unused-result")
  set_source_files_properties(fleet_py.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
  list(APPEND PYBIND_DEPS fleet communicator)
  list(APPEND PYBIND_SRCS fleet_py.cc)
62
endif()
63

64 65 66 67
if (WITH_NCCL)
  list(APPEND PYBIND_SRCS nccl_wrapper_py.cc)
endif()

L
Leo Chen 已提交
68 69 70 71 72 73
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})

74
  if(WITH_NCCL)
L
Leo Chen 已提交
75
    list(APPEND OP_FUNCTION_GENERETOR_DEPS nccl_context)
76
  endif(WITH_NCCL)
L
Leo Chen 已提交
77 78

  add_executable(op_function_generator op_function_generator.cc)
79
  target_link_libraries(op_function_generator ${OP_FUNCTION_GENERETOR_DEPS})
L
Leo Chen 已提交
80 81 82
  get_property (os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
  target_link_libraries(op_function_generator ${os_dependency_modules})

83 84 85
  set(impl_file ${CMAKE_SOURCE_DIR}/paddle/fluid/pybind/op_function_impl.h)
  set(tmp_impl_file ${impl_file}.tmp)

86
  if(WIN32)
87 88 89 90 91 92 93 94 95 96 97 98 99 100 101
    file(WRITE ${CMAKE_BINARY_DIR}/paddle/fluid/pybind/op_function_generator_retry.bat ""
    "set build_times=1\n"
    ":retry\n"
    "ECHO op_function_generator run %build_times% time\n"
    "${CMAKE_BINARY_DIR}/paddle/fluid/pybind/${CMAKE_BUILD_TYPE}/op_function_generator ${impl_file}\n"
    "if %ERRORLEVEL% NEQ 0 (\n"
    "    set /a build_times=%build_times%+1\n"
    "    if %build_times% GTR 100 (\n"
    "        exit /b 1\n"
    "    ) else (\n"
    "        goto :retry\n"
    "    )\n"
    ")\n"
    "exit /b 0")

Z
Zhou Wei 已提交
102
    add_custom_command(TARGET op_function_generator POST_BUILD
103
          COMMAND ${CMAKE_BINARY_DIR}/paddle/fluid/pybind/op_function_generator_retry.bat
104 105
    )

106
    if(${CBLAS_PROVIDER} STREQUAL MKLML)
107
      add_custom_command(TARGET op_function_generator
108
            PRE_LINK
109 110
            COMMAND ${CMAKE_COMMAND} -E copy ${MKLML_SHARED_LIB} ${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 已提交
111
          )
112
    else(${CBLAS_PROVIDER} STREQUAL EXTERN_OPENBLAS)
113
      add_custom_command(TARGET op_function_generator
114
            PRE_LINK
115 116 117
            COMMAND ${CMAKE_COMMAND} -E copy ${OPENBLAS_SHARED_LIB} ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_BUILD_TYPE}
          )
    endif()
L
Leo Chen 已提交
118
    if(WITH_MKLDNN)
119
      add_custom_command(TARGET op_function_generator
120
          PRE_LINK
L
Leo Chen 已提交
121 122
          COMMAND ${CMAKE_COMMAND} -E copy ${MKLDNN_SHARED_LIB} ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_BUILD_TYPE}
          )
123
    endif()
L
Leo Chen 已提交
124 125 126 127 128
  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.
129 130 131
    add_custom_command(TARGET op_function_generator
          POST_BUILD
          COMMAND ${CMAKE_COMMAND} -E env "LD_LIBRARY_PATH=$ENV{LD_LIBRARY_PATH}:."
132
              "${CMAKE_CURRENT_BINARY_DIR}/op_function_generator"
133 134 135 136 137
              "${tmp_impl_file}"
          COMMAND ${CMAKE_COMMAND} -E copy_if_different ${tmp_impl_file} ${impl_file}
          COMMENT "copy_if_different ${impl_file}"
          VERBATIM
    )
L
Leo Chen 已提交
138
    if(WITH_MKL)
139
      add_custom_command(TARGET op_function_generator
140
            PRE_LINK
141 142 143
            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 已提交
144 145
    endif(WITH_MKL)
    if(WITH_MKLDNN)
146
      add_custom_command(TARGET op_function_generator
147
          PRE_LINK
L
Leo Chen 已提交
148 149 150 151 152
          COMMAND ${CMAKE_COMMAND} -E copy ${MKLDNN_SHARED_LIB} ${CMAKE_CURRENT_BINARY_DIR}
          )
    endif(WITH_MKLDNN)
  endif(WIN32)

Y
Y_Xuan 已提交
153 154
  if(WITH_ROCM_PLATFORM)
    cc_library(paddle_pybind SHARED
D
dzhwinter 已提交
155
      SRCS ${PYBIND_SRCS}
Y
Y_Xuan 已提交
156 157
      DEPS ${PYBIND_DEPS}
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
158 159
  else()
    cc_library(paddle_pybind SHARED
D
dzhwinter 已提交
160
      SRCS ${PYBIND_SRCS}
D
dzhwinter 已提交
161
      DEPS ${PYBIND_DEPS}
W
Wu Yi 已提交
162
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
T
Tao Luo 已提交
163
    if(NOT APPLE AND NOT WIN32)
164
      target_link_libraries(paddle_pybind rt)
T
Tao Luo 已提交
165
    endif(NOT APPLE AND NOT WIN32)
Y
Y_Xuan 已提交
166
  endif(WITH_ROCM_PLATFORM)
167

P
peizhilin 已提交
168 169
  get_property (os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
  target_link_libraries(paddle_pybind ${os_dependency_modules})
170
  add_dependencies(paddle_pybind op_function_generator)
171
endif(WITH_PYTHON)