CMakeLists.txt 2.8 KB
Newer Older
1

2 3 4
file(GLOB TRAINER_PY_FILES . ./paddle/trainer/*.py)
file(GLOB HELPERS_PY_FILES . ./paddle/trainer_config_helpers/*.py)
file(GLOB UTILS_PY_FILES . ./paddle/utils/*.py)
Y
Yu Yang 已提交
5
file(GLOB_RECURSE V2_PY_FILES ./paddle/v2/ *.py)
6 7

set(PY_FILES paddle/__init__.py
H
Helin Wang 已提交
8 9 10 11 12 13
  ${TRAINER_PY_FILES}
  ${HELPERS_PY_FILES}
  ${UTILS_PY_FILES}
  ${V2_PY_FILES})

add_custom_target(copy_paddle_master)
D
dongzhihong 已提交
14 15

SET(COPY_PADDLE_MASTER "")
H
Helin Wang 已提交
16
if(WITH_GOLANG)
D
dongzhihong 已提交
17 18
  SET(COPY_PADDLE_MASTER "copy_paddle_master")
  add_custom_command(TARGET ${COPY_PADDLE_MASTER}
19
    COMMAND cp ${paddle_master_LIB_PATH} ${PADDLE_SOURCE_DIR}/python/paddle/v2/master/
H
Helin Wang 已提交
20 21 22
    )
  add_dependencies(copy_paddle_master paddle_master)
endif(WITH_GOLANG)
Z
zhangjinchao01 已提交
23

T
tensor-tang 已提交
24 25 26 27 28 29 30 31 32 33 34 35
set(MKL_SHARED_LIBS "")
set(MKL_DEPENDS "")
if(WITH_MKLML)
  list(APPEND MKL_SHARED_LIBS ${MKLML_LIB} ${MKLML_IOMP_LIB})
  list(APPEND MKL_DEPENDS mklml)
endif()

if(WITH_MKLDNN)
  list(APPEND MKL_SHARED_LIBS "${MKLDNN_LIB}" "${MKLDNN_LIB}.0")
  list(APPEND MKL_DEPENDS mkldnn)
endif()

T
typhoonzero 已提交
36 37 38 39 40 41
if(WITH_GPU)
  SET(PACKAGE_NAME "paddlepaddle_gpu")
else()
  SET(PACKAGE_NAME "paddlepaddle")
endif()

Z
zhangjinchao01 已提交
42 43 44
configure_file(${CMAKE_CURRENT_SOURCE_DIR}/setup.py.in
    ${CMAKE_CURRENT_BINARY_DIR}/setup.py)

45

Q
Qiao Longfei 已提交
46 47
add_custom_command(OUTPUT ${PADDLE_SOURCE_DIR}/python/paddle/v2/fluid/core.so
        COMMAND cmake -E copy $<TARGET_FILE:paddle_pybind> ${PADDLE_SOURCE_DIR}/python/paddle/v2/fluid/core.so
48
        DEPENDS paddle_pybind)
Q
Qiao Longfei 已提交
49
add_custom_target(copy_paddle_pybind ALL DEPENDS ${PADDLE_SOURCE_DIR}/python/paddle/v2/fluid/core.so)
50 51


Q
qiaolongfei 已提交
52
add_custom_command(OUTPUT ${PADDLE_PYTHON_BUILD_DIR}/.timestamp
T
typhoonzero 已提交
53
    COMMAND touch stub.cc
L
liaogang 已提交
54
    COMMAND env ${py_env} ${PYTHON_EXECUTABLE} setup.py bdist_wheel
Q
qiaolongfei 已提交
55
    COMMAND ${CMAKE_COMMAND} -E touch ${PADDLE_PYTHON_BUILD_DIR}/.timestamp
Q
qiaolongfei 已提交
56 57
    COMMAND ${CMAKE_COMMAND} -E remove_directory ${PADDLE_PYTHON_BUILD_DIR}/lib-python
    COMMAND ${CMAKE_COMMAND} -E copy_directory ${PADDLE_PYTHON_BUILD_DIR}/lib* ${PADDLE_PYTHON_BUILD_DIR}/lib-python
58
    DEPENDS gen_proto_py copy_paddle_pybind framework_py_proto ${PY_FILES} ${external_project_dependencies} ${COPY_PADDLE_MASTER})
Z
zhangjinchao01 已提交
59

60 61 62 63 64
set(paddle_python_deps ${PADDLE_PYTHON_BUILD_DIR}/.timestamp paddle_pserver_main paddle_trainer paddle_merge_model ${MKL_DEPENDS})
if(WITH_SWIG_PY)
    list(APPEND paddle_python_deps python_api_wheel)
endif()
add_custom_target(paddle_python ALL DEPENDS ${paddle_python_deps})
Z
zhangjinchao01 已提交
65

Y
Yu Yang 已提交
66 67
set(PADDLE_PYTHON_PACKAGE_DIR ${CMAKE_CURRENT_BINARY_DIR}/dist/)

68 69 70 71 72 73 74
if (WITH_TESTING)
  add_subdirectory(paddle/trainer_config_helpers/tests)
  if (WITH_SWIG_PY)
    # enable v2 API unittest only when paddle swig api is compiled
    add_subdirectory(paddle/v2/tests)
    add_subdirectory(paddle/v2/reader/tests)
    add_subdirectory(paddle/v2/plot/tests)
Q
Qiao Longfei 已提交
75
    add_subdirectory(paddle/v2/fluid/tests)
76
  endif()
77
endif()
Y
Yu Yang 已提交
78
install(DIRECTORY ${PADDLE_PYTHON_PACKAGE_DIR}
Z
zhangjinchao01 已提交
79 80
    DESTINATION opt/paddle/share/wheels
)