CMakeLists.txt 774 字节
Newer Older
D
dzhwinter 已提交
1 2
set(PYBIND_DEPS pybind python proto_desc memory executor prune  feed_fetch_method)
set(PYBIND_SRCS pybind.cc exception.cc protobuf.cc const_value.cc)
D
dzhwinter 已提交
3
if(NOT WIN32)
D
dzhwinter 已提交
4 5
list(APPEND PYBIND_DEPS parallel_executor profiler)
list(APPEND PYBIND_SRCS recordio.cc)
D
dzhwinter 已提交
6
endif()
7
if(WITH_PYTHON)
8 9
  if(WITH_AMD_GPU)
    hip_library(paddle_pybind SHARED
D
dzhwinter 已提交
10
      SRCS ${PYBIND_SRCS}
D
dzhwinter 已提交
11
      DEPS ${PYBIND_DEPS}
12 13 14
      ${GLOB_OP_LIB})
  else()
    cc_library(paddle_pybind SHARED
D
dzhwinter 已提交
15
      SRCS ${PYBIND_SRCS}
D
dzhwinter 已提交
16
      DEPS ${PYBIND_DEPS}
17
      ${GLOB_OP_LIB})
D
dzhwinter 已提交
18
    if(NOT APPLE AND NOT ANDROID AND NOT WIN32)
19
      target_link_libraries(paddle_pybind rt)
D
dzhwinter 已提交
20
    endif(NOT APPLE AND NOT ANDROID AND NOT WIN32)
21
  endif(WITH_AMD_GPU)
22 23

  cc_test(tensor_py_test SRCS tensor_py_test.cc DEPS python)
24
endif(WITH_PYTHON)