CMakeLists.txt 1.3 KB
Newer Older
D
dongdaxiang 已提交
1
set(PYBIND_DEPS pybind python proto_desc memory executor async_executor fleet_wrapper prune
M
minqiyang 已提交
2
  feed_fetch_method pass_builder parallel_executor profiler layer scope_pool
M
minqiyang 已提交
3
  tracer analysis_predictor imperative_profiler)
F
flame 已提交
4

S
sneaxiy 已提交
5 6 7
if(WITH_PYTHON)
  list(APPEND PYBIND_DEPS py_func_op)
endif()
D
dongdaxiang 已提交
8
<<<<<<< HEAD
S
sneaxiy 已提交
9
set(PYBIND_SRCS pybind.cc exception.cc protobuf.cc const_value.cc recordio.cc reader_py.cc async_executor_py.cc imperative.cc ir.cc inference_api.cc)
D
dongdaxiang 已提交
10 11 12
=======
set(PYBIND_SRCS pybind.cc exception.cc protobuf.cc const_value.cc recordio.cc async_executor_py.cc fleet_wrapper_py.cc imperative.cc ir.cc inference_api.cc)
>>>>>>> add pybind for fleet
13

14
if(WITH_PYTHON)
15 16
  if(WITH_AMD_GPU)
    hip_library(paddle_pybind SHARED
D
dzhwinter 已提交
17
      SRCS ${PYBIND_SRCS}
S
sabreshao 已提交
18 19
      DEPS ARCHIVE_START ${PYBIND_DEPS}
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS} ARCHIVE_END)
20 21
  else()
    cc_library(paddle_pybind SHARED
D
dzhwinter 已提交
22
      SRCS ${PYBIND_SRCS}
D
dzhwinter 已提交
23
      DEPS ${PYBIND_DEPS}
W
Wu Yi 已提交
24
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
T
Tao Luo 已提交
25
    if(NOT APPLE AND NOT WIN32)
26
      target_link_libraries(paddle_pybind rt)
T
Tao Luo 已提交
27
    endif(NOT APPLE AND NOT WIN32)
28
  endif(WITH_AMD_GPU)
29

P
peizhilin 已提交
30 31
  get_property (os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
  target_link_libraries(paddle_pybind ${os_dependency_modules})
P
peizhilin 已提交
32

33
  cc_test(tensor_py_test SRCS tensor_py_test.cc DEPS python pybind)
34
endif(WITH_PYTHON)