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

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