CMakeLists.txt 17.3 KB
Newer Older
1 2
set(PYBIND_DEPS
    init
3
    init_phi
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
    pybind
    python
    proto_desc
    memory
    executor
    fleet_wrapper
    box_wrapper
    metrics
    prune
    feed_fetch_method
    pass
    generate_pass
    pass_builder
    parallel_executor
    profiler
    layer
    tracer
    engine
    scope_pool
    analysis_predictor
    imperative_profiler
    imperative_flag
    dlpack_tensor
    device_context
    gloo_wrapper
    infer_io_utils
    heter_wrapper
    generator
    op_version_registry
    ps_gpu_wrapper
    custom_operator
    cost_model
    cuda_graph_with_memory_pool
    fleet_executor
    global_utils
    phi_utils
    tcp_store
41
    comm_context_manager
42
    new_profiler
43
    auto_parallel
44
    jit_layer
J
Jiabin Yang 已提交
45
    jit_property
46 47 48 49
    prim_utils
    operants_manager
    phi_tensor_operants
    static_tensor_operants)
50 51

if(WITH_PSCORE)
S
seemingwang 已提交
52
  set(PYBIND_DEPS ${PYBIND_DEPS} ps_service)
53
  if(WITH_HETERPS)
54 55
    set(PYBIND_DEPS ${PYBIND_DEPS} graph_gpu_wrapper)
  endif()
S
seemingwang 已提交
56
endif()
X
Xinger 已提交
57
if(WITH_RPC)
58 59 60 61 62 63 64 65 66 67 68 69
  set(PYBIND_DEPS
      ${PYBIND_DEPS}
      paddle_rpc
      brpc
      ssl
      crypto
      protobuf
      zlib
      leveldb
      snappy
      gflags
      glog)
70
endif()
71
if(WITH_GPU OR WITH_ROCM)
72
  set(PYBIND_DEPS ${PYBIND_DEPS} dynload_cuda)
73
  set(PYBIND_DEPS ${PYBIND_DEPS} cuda_device_guard)
74 75
endif()

76
if(WITH_GPU)
77 78
  set(PYBIND_DEPS ${PYBIND_DEPS} cuda_profiler)
endif()
79
if(WITH_IPU)
J
jianghaicheng 已提交
80 81
  set(PYBIND_DEPS ${PYBIND_DEPS} ipu_info)
endif()
82

83
if(WITH_NCCL OR WITH_RCCL)
84
  set(PYBIND_DEPS ${PYBIND_DEPS} nccl_wrapper)
85
  set(PYBIND_DEPS ${PYBIND_DEPS} reducer)
86 87
endif()

88
if(WITH_XPU_BKCL)
89 90
  set(PYBIND_DEPS ${PYBIND_DEPS} reducer)
  set(PYBIND_DEPS ${PYBIND_DEPS} bkcl_context)
K
kuizhiqing 已提交
91 92 93
  set(PYBIND_DEPS ${PYBIND_DEPS} heter_ccl_context)
endif()

94 95 96 97 98 99 100 101 102 103 104
if(WITH_CUSTOM_DEVICE)
  set(PYBIND_DEPS ${PYBIND_DEPS} xccl_context)
  if(NOT
     (WITH_NCCL
      OR WITH_RCCL
      OR WITH_XPU_BKCL))
    set(PYBIND_DEPS ${PYBIND_DEPS} reducer)
    set(PYBIND_DEPS ${PYBIND_DEPS} heter_ccl_context)
  endif()
endif()

105
if(NOT WIN32)
106
  set(PYBIND_DEPS ${PYBIND_DEPS} data_loader)
107
  if(WITH_NCCL OR WITH_RCCL)
108
    set(PYBIND_DEPS ${PYBIND_DEPS} nccl_context)
K
kuizhiqing 已提交
109
    set(PYBIND_DEPS ${PYBIND_DEPS} heter_ccl_context)
110
  endif()
111
endif()
F
flame 已提交
112

S
sneaxiy 已提交
113 114 115
if(WITH_PYTHON)
  list(APPEND PYBIND_DEPS py_func_op)
endif()
116 117

set(PYBIND_SRCS
118
    pybind.cc
119 120 121 122 123
    imperative.cc
    inference_api.cc
    ir.cc
    bind_fleet_executor.cc
    reader_py.cc
124
    protobuf.cc
125
    exception.cc
126 127 128 129
    op_function_common.cc
    parallel_executor.cc
    tensor.cc
    place.cc
130 131 132 133 134 135 136 137 138 139 140 141 142 143
    const_value.cc
    global_value_getter_setter.cc
    fleet_wrapper_py.cc
    heter_wrapper_py.cc
    ps_gpu_wrapper_py.cc
    gloo_wrapper_py.cc
    box_helper_py.cc
    metrics_py.cc
    data_set_py.cc
    bind_cost_model.cc
    compatible.cc
    io.cc
    generator_py.cc
    communication.cc
144
    cuda_streams_py.cc
145
    custom_device_py.cc
J
james 已提交
146
    xpu_streams_py.cc
147
    jit.cc
姜永久 已提交
148
    auto_parallel_py.cc)
149

150 151
if(WITH_CUSTOM_DEVICE)
  set(PYBIND_DEPS ${PYBIND_DEPS} phi_capi)
152
  set(PYBIND_DEPS ${PYBIND_DEPS} custom_device_common_op_registry)
153 154
endif()

155
if(WITH_PYTHON)
W
Wen Sun 已提交
156
  set(PYBIND_DEPS ${PYBIND_DEPS} process_group eager_reducer)
157
  if(WITH_NCCL OR WITH_RCCL)
W
Wen Sun 已提交
158
    set(PYBIND_DEPS ${PYBIND_DEPS} process_group_nccl)
159
  endif()
J
james 已提交
160
  if(WITH_XPU_BKCL)
W
Wen Sun 已提交
161
    set(PYBIND_DEPS ${PYBIND_DEPS} process_group_bkcl)
J
james 已提交
162
  endif()
163
  if(WITH_GLOO)
W
Wen Sun 已提交
164
    set(PYBIND_DEPS ${PYBIND_DEPS} process_group_gloo)
165
  endif()
W
wuhuachaocoding 已提交
166
  if(WITH_MPI)
W
Wen Sun 已提交
167
    set(PYBIND_DEPS ${PYBIND_DEPS} process_group_mpi)
W
wuhuachaocoding 已提交
168
  endif()
169
  if(WITH_CUSTOM_DEVICE)
W
Wen Sun 已提交
170
    set(PYBIND_DEPS ${PYBIND_DEPS} process_group_custom)
171
  endif()
172 173 174 175 176
  if(CMAKE_CXX_COMPILER_VERSION VERSION_GREATER 7.0)
    set(DISTRIBUTE_COMPILE_FLAGS "${DISTRIBUTE_COMPILE_FLAGS} -faligned-new")
    set_source_files_properties(
      distributed_py.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
  endif()
177 178
  set(PYBIND_SRCS ${PYBIND_SRCS} distributed_py.cc)
endif()
179
set(PYBIND_DEPS ${PYBIND_DEPS} processgroup_comm_utils)
180

181 182 183
if(WITH_GLOO)
  set(PYBIND_DEPS ${PYBIND_DEPS} gloo_context)
  set(PYBIND_SRCS ${PYBIND_SRCS} gloo_context_py.cc)
184 185
  set(PYBIND_DEPS ${PYBIND_DEPS} imperative_gloo_context)
  set(PYBIND_DEPS ${PYBIND_DEPS} reducer)
186
endif()
187

188
if(WITH_CRYPTO)
Y
Yanghello 已提交
189 190
  set(PYBIND_DEPS ${PYBIND_DEPS} paddle_crypto)
  set(PYBIND_SRCS ${PYBIND_SRCS} crypto.cc)
191
endif()
192 193 194

if(WITH_PSLIB)
  set(DISTRIBUTE_COMPILE_FLAGS
195
      "-Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor -Wno-error=return-type -Wno-error=unused-but-set-variable -Wno-error=type-limits -Wno-error=parentheses -Wno-error=unused-result"
196 197 198
  )
  if(CMAKE_CXX_COMPILER_VERSION VERSION_GREATER 7.0)
    set(DISTRIBUTE_COMPILE_FLAGS "${DISTRIBUTE_COMPILE_FLAGS} -faligned-new")
T
Thunderbrook 已提交
199
  endif()
200 201
  set_source_files_properties(
    heter_wrapper_py.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
202
endif()
203 204 205
if(WITH_PSCORE)
  if(WITH_ARM_BRPC)
    set(DISTRIBUTE_COMPILE_FLAGS
206
        "-faligned-new -Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor -Wno-error=return-type -Wno-error=unused-but-set-variable -Wno-error=parentheses -Wno-error=unused-result"
207
    )
208
  else()
209
    set(DISTRIBUTE_COMPILE_FLAGS
210
        "-Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor -Wno-error=return-type -Wno-error=unused-but-set-variable -Wno-error=parentheses -Wno-error=unused-result"
211
    )
212 213 214
    if(CMAKE_CXX_COMPILER_VERSION VERSION_GREATER 7.0)
      set(DISTRIBUTE_COMPILE_FLAGS "${DISTRIBUTE_COMPILE_FLAGS} -faligned-new")
    endif()
215
  endif()
216 217
  set_source_files_properties(
    fleet_py.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
Z
zhaocaibei123 已提交
218
  list(APPEND PYBIND_DEPS fleet index_wrapper index_sampler)
219 220
  list(APPEND PYBIND_SRCS)
  set(PYBIND_SRCS fleet_py.cc ${PYBIND_SRCS})
221
endif()
222

X
Xinger 已提交
223
if(WITH_RPC)
224 225
  if(WITH_ARM_BRPC)
    set(DISTRIBUTE_COMPILE_FLAGS
226
        "-faligned-new -Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor -Wno-error=return-type -Wno-error=unused-but-set-variable -Wno-error=parentheses -Wno-error=unused-result"
227 228 229
    )
  else()
    set(DISTRIBUTE_COMPILE_FLAGS
230
        "-Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor -Wno-error=return-type -Wno-error=unused-but-set-variable -Wno-error=parentheses -Wno-error=unused-result"
231 232 233 234 235 236 237 238 239 240
    )
    if(CMAKE_CXX_COMPILER_VERSION VERSION_GREATER 7.0)
      set(DISTRIBUTE_COMPILE_FLAGS "${DISTRIBUTE_COMPILE_FLAGS} -faligned-new")
    endif()
  endif()
  set_source_files_properties(rpc.cc PROPERTIES COMPILE_FLAGS
                                                ${DISTRIBUTE_COMPILE_FLAGS})
  set(PYBIND_SRCS rpc.cc ${PYBIND_SRCS})
endif()

241
if(WITH_NCCL OR WITH_RCCL)
242 243 244
  list(APPEND PYBIND_SRCS nccl_wrapper_py.cc)
endif()

L
Leo Chen 已提交
245 246
if(WITH_PYTHON)
  # generate op pybind functions automatically for dygraph.
247 248 249 250 251 252 253 254 255 256

  set(OP_FUNCTION_GENERETOR_DEPS
      pybind
      proto_desc
      executor
      layer
      tracer
      engine
      imperative_profiler
      imperative_flag)
L
Leo Chen 已提交
257 258 259
  list(APPEND OP_FUNCTION_GENERETOR_DEPS ${GLOB_OP_LIB})
  list(APPEND OP_FUNCTION_GENERETOR_DEPS ${GLOB_OPERATOR_DEPS})

260
  if(WITH_NCCL OR WITH_RCCL)
L
Leo Chen 已提交
261
    list(APPEND OP_FUNCTION_GENERETOR_DEPS nccl_context)
262
  endif()
L
Leo Chen 已提交
263

264 265
  if(WITH_XPU_BKCL)
    list(APPEND OP_FUNCTION_GENERETOR_DEPS bkcl_context)
266
  endif()
267

268 269 270 271
  if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
    list(APPEND OP_FUNCTION_GENERETOR_DEPS ${PYTHON_LIBRARIES})
  endif()

272 273 274 275 276
  if(WITH_CUSTOM_DEVICE)
    set(OP_FUNCTION_GENERETOR_DEPS ${OP_FUNCTION_GENERETOR_DEPS}
                                   custom_device_common_op_registry)
  endif()

277 278 279
  add_executable(eager_legacy_op_function_generator
                 eager_legacy_op_function_generator.cc)
  target_link_libraries(eager_legacy_op_function_generator
280
                        ${OP_FUNCTION_GENERETOR_DEPS})
281 282
  if(NOT WIN32)
    add_executable(kernel_signature_generator kernel_signature_generator.cc)
283 284
    target_link_libraries(kernel_signature_generator
                          ${OP_FUNCTION_GENERETOR_DEPS})
285
  endif()
286

287
  get_property(os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
288 289
  target_link_libraries(eager_legacy_op_function_generator
                        ${os_dependency_modules})
290
  if(WITH_ROCM)
291
    target_link_libraries(eager_legacy_op_function_generator ${ROCM_HIPRTC_LIB})
292
    target_link_libraries(kernel_signature_generator ${ROCM_HIPRTC_LIB})
293
  endif()
L
Leo Chen 已提交
294

295 296
  set(op_function_output_path ${CMAKE_SOURCE_DIR}/paddle/fluid/pybind/)
  set(CODE_GEN_SPLIT_FILE_COUNT "8")
297
  set(eager_impl_file
298
      ${CMAKE_SOURCE_DIR}/paddle/fluid/pybind/eager_legacy_op_function.cc)
299
  set(tmp_eager_impl_file ${eager_impl_file}.tmp)
300

301 302 303 304 305 306 307
  execute_process(
    COMMAND
      "${PYTHON_EXECUTABLE}"
      "${PADDLE_SOURCE_DIR}/paddle/fluid/pybind/generate_file_structures.py"
      "${PADDLE_SOURCE_DIR}/paddle/fluid/pybind/"
      "${CODE_GEN_SPLIT_FILE_COUNT}")

308 309
  set(EAGER_OP_IMPL_DEPS eager_legacy_op_function_generator
                         eager_python_c_codegen)
310

311
  if(WIN32)
312
    if("${CMAKE_GENERATOR}" STREQUAL "Ninja")
313
      set(op_impl_path "${CMAKE_CURRENT_BINARY_DIR}")
314
    else()
315
      set(op_impl_path "${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_BUILD_TYPE}")
316
    endif()
317

318 319
    file(
      WRITE
320
      ${CMAKE_BINARY_DIR}/paddle/fluid/pybind/eager_legacy_op_function_generator_retry.bat
321 322 323
      ""
      "set build_times=1\n"
      ":retry\n"
324 325 326
      "ECHO eager_legacy_op_function_generator run %build_times% time\n"
      "taskkill /f /im eager_legacy_op_function_generator.exe 2>NUL\n"
      "${op_impl_path}/eager_legacy_op_function_generator.exe ${tmp_eager_impl_file}\n"
327 328 329 330 331 332 333 334 335
      "if %ERRORLEVEL% NEQ 0 (\n"
      "    set /a build_times=%build_times%+1\n"
      "    if %build_times% GEQ 10 (\n"
      "        exit /b 1\n"
      "    ) else (\n"
      "        goto :retry\n"
      "    )\n"
      ")\n"
      "exit /b 0")
336

337
    if(${CBLAS_PROVIDER} STREQUAL MKLML)
338 339 340 341
      add_custom_command(
        OUTPUT ${op_impl_path}/libiomp5md.dll
        COMMAND ${CMAKE_COMMAND} -E copy ${MKLML_SHARED_IOMP_LIB}
                ${op_impl_path}
342 343
        DEPENDS mklml)
      list(APPEND OP_IMPL_DEPS ${op_impl_path}/libiomp5md.dll)
344
      list(APPEND EAGER_OP_IMPL_DEPS ${op_impl_path}/libiomp5md.dll)
345
    else()
346 347
      add_custom_command(
        OUTPUT ${op_impl_path}/openblas.dll
348 349 350
        COMMAND ${CMAKE_COMMAND} -E copy ${OPENBLAS_SHARED_LIB} ${op_impl_path}
        DEPENDS extern_openblas)
      list(APPEND OP_IMPL_DEPS ${op_impl_path}/openblas.dll)
351
      list(APPEND EAGER_OP_IMPL_DEPS ${op_impl_path}/openblas.dll)
352
    endif()
L
Leo Chen 已提交
353
    if(WITH_MKLDNN)
354 355
      add_custom_command(
        OUTPUT ${op_impl_path}/mkldnn.dll
356 357
        COMMAND ${CMAKE_COMMAND} -E copy ${MKLDNN_SHARED_LIB} ${op_impl_path}
        DEPENDS mkldnn)
358 359
      list(APPEND OP_IMPL_DEPS ${op_impl_path}/mkldnn.dll)
      list(APPEND EAGER_OP_IMPL_DEPS ${op_impl_path}/mkldnn.dll)
360
    endif()
361
    if(WITH_ONNXRUNTIME)
362 363
      add_custom_command(
        OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/paddle2onnx.dll
364
        COMMAND ${CMAKE_COMMAND} -E copy ${PADDLE2ONNX_LIB}
365
                ${CMAKE_CURRENT_BINARY_DIR}
366 367
        DEPENDS paddle2onnx)
      list(APPEND OP_IMPL_DEPS ${CMAKE_CURRENT_BINARY_DIR}/paddle2onnx.dll)
368 369
      list(APPEND EAGER_OP_IMPL_DEPS
           ${CMAKE_CURRENT_BINARY_DIR}/paddle2onnx.dll)
370

371 372 373 374
      add_custom_command(
        OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/onnxruntime.dll
        COMMAND ${CMAKE_COMMAND} -E copy ${ONNXRUNTIME_SHARED_LIB}
                ${CMAKE_CURRENT_BINARY_DIR}
375 376
        DEPENDS onnxruntime)
      list(APPEND OP_IMPL_DEPS ${CMAKE_CURRENT_BINARY_DIR}/onnxruntime.dll)
377 378
      list(APPEND EAGER_OP_IMPL_DEPS
           ${CMAKE_CURRENT_BINARY_DIR}/onnxruntime.dll)
379
    endif()
380

381
    if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
382 383 384
      add_custom_command(
        OUTPUT ${eager_impl_file}
        COMMAND
385
          ${CMAKE_BINARY_DIR}/paddle/fluid/pybind/eager_legacy_op_function_generator_retry.bat
386 387
        COMMAND ${CMAKE_COMMAND} -E copy_if_different ${tmp_eager_impl_file}
                ${eager_impl_file}
388 389 390
        COMMENT "copy_if_different ${tmp_eager_impl_file} to ${eager_impl_file}"
        DEPENDS ${EAGER_OP_IMPL_DEPS})
    endif()
391
  else()
392
    # If there are no *.so in /usr/lib or LD_LIBRARY_PATH,
L
Leo Chen 已提交
393
    # copy these *.so to current directory and append current directory to
394
    # LD_LIBRARY_PATH. This is different with Windows platformm, which search
L
Leo Chen 已提交
395
    # *.dll in current directory automatically.
396
    if(WITH_ONNXRUNTIME)
H
heliqi 已提交
397 398 399 400
      set(PADDLE2ONNX_PYBIND_OUT
          ${CMAKE_CURRENT_BINARY_DIR}/${PADDLE2ONNX_LIB_NAME})
      set(ONNXRUNTIME_PYBIND_OUT
          ${CMAKE_CURRENT_BINARY_DIR}/${ONNXRUNTIME_LIB_NAME})
401

402 403 404 405
      add_custom_command(
        OUTPUT ${PADDLE2ONNX_PYBIND_OUT}
        COMMAND ${CMAKE_COMMAND} -E copy ${PADDLE2ONNX_LIB}
                ${CMAKE_CURRENT_BINARY_DIR}
406 407 408 409
        DEPENDS paddle2onnx)
      list(APPEND OP_IMPL_DEPS ${PADDLE2ONNX_PYBIND_OUT})
      list(APPEND EAGER_OP_IMPL_DEPS ${PADDLE2ONNX_PYBIND_OUT})

410 411 412 413
      add_custom_command(
        OUTPUT ${ONNXRUNTIME_PYBIND_OUT}
        COMMAND ${CMAKE_COMMAND} -E copy ${ONNXRUNTIME_LIB}
                ${CMAKE_CURRENT_BINARY_DIR}
414 415 416 417 418
        DEPENDS onnxruntime)
      list(APPEND OP_IMPL_DEPS ${ONNXRUNTIME_PYBIND_OUT})
      list(APPEND EAGER_OP_IMPL_DEPS ${ONNXRUNTIME_PYBIND_OUT})
    endif()

419
    if(WITH_MKLML)
420 421 422 423
      add_custom_command(
        OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/libiomp5.so
        COMMAND ${CMAKE_COMMAND} -E copy ${MKLML_SHARED_IOMP_LIB}
                ${CMAKE_CURRENT_BINARY_DIR}
424 425
        DEPENDS mklml)
      list(APPEND OP_IMPL_DEPS ${CMAKE_CURRENT_BINARY_DIR}/libiomp5.so)
426
      list(APPEND EAGER_OP_IMPL_DEPS ${CMAKE_CURRENT_BINARY_DIR}/libiomp5.so)
427 428
    endif()
    if(WITH_MKLDNN)
429 430 431 432
      add_custom_command(
        OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/libdnnl.so.0
        COMMAND ${CMAKE_COMMAND} -E copy ${MKLDNN_SHARED_LIB}
                ${CMAKE_CURRENT_BINARY_DIR}
433 434
        DEPENDS mkldnn)
      list(APPEND OP_IMPL_DEPS ${CMAKE_CURRENT_BINARY_DIR}/libdnnl.so.0)
435
      list(APPEND EAGER_OP_IMPL_DEPS ${CMAKE_CURRENT_BINARY_DIR}/libdnnl.so.0)
436
    endif()
437
    if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
438 439 440 441
      add_custom_command(
        OUTPUT ${eager_impl_file}
        COMMAND
          ${CMAKE_COMMAND} -E env "LD_LIBRARY_PATH=$ENV{LD_LIBRARY_PATH}:."
442
          "${CMAKE_CURRENT_BINARY_DIR}/eager_legacy_op_function_generator"
443 444 445 446 447 448 449
          "${tmp_eager_impl_file}"
        COMMAND ${CMAKE_COMMAND} -E copy_if_different ${tmp_eager_impl_file}
                ${eager_impl_file}
        COMMENT "copy_if_different ${tmp_eager_impl_file} to ${eager_impl_file}"
        DEPENDS ${EAGER_OP_IMPL_DEPS}
        VERBATIM)
    endif()
450
  endif()
451
  if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
452
    add_custom_target(eager_legacy_op_function_generator_cmd ALL
453
                      DEPENDS ${eager_impl_file})
454
  endif()
L
Leo Chen 已提交
455

456
  list(APPEND PYBIND_DEPS standalone_executor staticgraph_executor_statistics)
457

458
  if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
459 460 461 462 463 464
    set(PYBIND_SRCS eager.cc ${PYBIND_SRCS})
    set(PYBIND_SRCS eager_functions.cc ${PYBIND_SRCS})
    set(PYBIND_SRCS eager_method.cc ${PYBIND_SRCS})
    set(PYBIND_SRCS eager_properties.cc ${PYBIND_SRCS})
    set(PYBIND_SRCS eager_utils.cc ${PYBIND_SRCS})
    set(PYBIND_SRCS eager_py_layer.cc ${PYBIND_SRCS})
465
    set(PYBIND_SRCS eager_legacy_op_function.cc ${PYBIND_SRCS})
466
    set(PYBIND_SRCS eager_op_function.cc ${PYBIND_SRCS})
467
    set(PYBIND_SRCS eager_math_op_patch.cc ${PYBIND_SRCS})
468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483
    list(APPEND PYBIND_DEPS eager_api)
    list(APPEND PYBIND_DEPS autograd_meta)
    list(APPEND PYBIND_DEPS backward)
    list(APPEND PYBIND_DEPS grad_node_info)
    list(APPEND PYBIND_DEPS phi)
    list(APPEND PYBIND_DEPS final_dygraph_function)
    list(APPEND PYBIND_DEPS final_dygraph_node)
    list(APPEND PYBIND_DEPS dygraph_function)
    list(APPEND PYBIND_DEPS dygraph_node)
    list(APPEND PYBIND_DEPS accumulation_node)
    list(APPEND PYBIND_DEPS py_layer_node)
    list(APPEND PYBIND_DEPS global_utils)
    list(APPEND PYBIND_DEPS utils)
    list(APPEND PYBIND_DEPS python)
    list(APPEND PYBIND_DEPS custom_operator)
    list(APPEND PYBIND_DEPS custom_operator_node)
484
    list(APPEND PYBIND_DEPS tensor_api)
485
    list(APPEND PYBIND_DEPS eager_tensor_operants)
486
    list(APPEND PYBIND_DEPS pybind_util)
487
    list(APPEND PYBIND_DEPS flags)
488
  endif()
489

490 491 492 493 494 495 496 497
  # On Linux, cc_library(paddle SHARED ..) will generate the libpaddle.so,
  # add a prefix `lib` by default, but on Windows, cc_library(paddle SHARED ..)
  # will not add prefix, so it generate paddle.lib and paddle.pyd,
  # we need to pay attention to the difference
  set(SHARD_LIB_NAME paddle)
  if(WIN32)
    set(SHARD_LIB_NAME libpaddle)
  endif()
498
  set_property(GLOBAL PROPERTY PADDLE_LIB_NAME ${SHARD_LIB_NAME})
499
  cc_library(
500
    ${SHARD_LIB_NAME} SHARED
501
    SRCS ${PYBIND_SRCS}
502
    DEPS ${PYBIND_DEPS} ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
503

504 505 506 507 508 509 510 511
  # TODO(zhiqiu): some symbols not exported even setting the following
  # property. Need to find a better way.

  # if(WIN32)
  #   set_property(TARGET ${SHARD_LIB_NAME}
  #                PROPERTY WINDOWS_EXPORT_ALL_SYMBOLS ON)
  # endif()

512
  if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
513 514
    add_dependencies(${SHARD_LIB_NAME} legacy_eager_codegen)
    add_dependencies(${SHARD_LIB_NAME} eager_legacy_op_function_generator_cmd)
515 516
  endif()

517
  if(NOT APPLE AND NOT WIN32)
518
    target_link_libraries(${SHARD_LIB_NAME} rt)
519
  endif()
520 521

  if(WITH_ROCM)
522
    target_link_libraries(${SHARD_LIB_NAME} ${ROCM_HIPRTC_LIB})
523
  endif()
524

W
wuhuachaocoding 已提交
525
  if(WITH_MPI)
526
    target_link_libraries(${SHARD_LIB_NAME} ${MPI_CXX_LIBRARIES})
W
wuhuachaocoding 已提交
527 528
  endif()

529
  get_property(os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
530
  target_link_libraries(${SHARD_LIB_NAME} ${os_dependency_modules})
531

532
endif()