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

if(NOT WIN32)
  set(PYBIND_DEPS ${PYBIND_DEPS} nccl_context)
endif(NOT WIN32)
F
flame 已提交
8

S
sneaxiy 已提交
9 10 11
if(WITH_PYTHON)
  list(APPEND PYBIND_DEPS py_func_op)
endif()
12 13 14 15 16 17 18 19 20 21

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

set(PYBIND_SRCS
  pybind.cc
  exception.cc
  protobuf.cc
  const_value.cc
22
  global_value_getter_setter.cc
23 24
  reader_py.cc
  fleet_wrapper_py.cc
H
hutuxian 已提交
25
  box_helper_py.cc
26 27 28 29 30 31 32 33 34
  nccl_wrapper_py.cc
  data_set_py.cc
  imperative.cc
  ir.cc
  inference_api.cc)

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

36
if(WITH_PYTHON)
37 38
  if(WITH_AMD_GPU)
    hip_library(paddle_pybind SHARED
D
dzhwinter 已提交
39
      SRCS ${PYBIND_SRCS}
S
sabreshao 已提交
40 41
      DEPS ARCHIVE_START ${PYBIND_DEPS}
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS} ARCHIVE_END)
42 43
  else()
    cc_library(paddle_pybind SHARED
D
dzhwinter 已提交
44
      SRCS ${PYBIND_SRCS}
D
dzhwinter 已提交
45
      DEPS ${PYBIND_DEPS}
W
Wu Yi 已提交
46
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
T
Tao Luo 已提交
47
    if(NOT APPLE AND NOT WIN32)
48
      target_link_libraries(paddle_pybind rt)
T
Tao Luo 已提交
49
    endif(NOT APPLE AND NOT WIN32)
50
  endif(WITH_AMD_GPU)
51

P
peizhilin 已提交
52 53
  get_property (os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
  target_link_libraries(paddle_pybind ${os_dependency_modules})
P
peizhilin 已提交
54

55
endif(WITH_PYTHON)