CMakeLists.txt 1.9 KB
Newer Older
Z
zhangjinchao01 已提交
1 2 3
set(OUTPUT_DIR
    "${CMAKE_CURRENT_BINARY_DIR}/build")

4 5 6
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 已提交
7
file(GLOB_RECURSE V2_PY_FILES ./paddle/v2/ *.py)
8 9

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

add_custom_target(copy_paddle_master)
D
dongzhihong 已提交
16 17

SET(COPY_PADDLE_MASTER "")
H
Helin Wang 已提交
18
if(WITH_GOLANG)
D
dongzhihong 已提交
19
  SET(COPY_PADDLE_MASTER "copy_paddle_master")
G
gongweibao 已提交
20 21
  message("paddle_master_lib_path:" ${paddle_master_LIB_PATH})
  message("PROJ_ROOT:" ${PROJ_ROOT})
D
dongzhihong 已提交
22
  add_custom_command(TARGET ${COPY_PADDLE_MASTER}
H
Helin Wang 已提交
23 24 25 26
    COMMAND cp ${paddle_master_LIB_PATH} ${PROJ_ROOT}/python/paddle/v2/master/
    )
  add_dependencies(copy_paddle_master paddle_master)
endif(WITH_GOLANG)
Z
zhangjinchao01 已提交
27

G
gongweibao 已提交
28 29
message("paddle_master_LIB_NAME:" ${paddle_master_LIB_NAME})
message("CMAKE_CURRENT_BINARY_DIR:" ${CMAKE_CURRENT_BINARY_DIR})
Z
zhangjinchao01 已提交
30 31 32
configure_file(${CMAKE_CURRENT_SOURCE_DIR}/setup.py.in
    ${CMAKE_CURRENT_BINARY_DIR}/setup.py)

G
gongweibao 已提交
33 34
message("OUTPUT_DIR:" ${OUTPUT_DIR})
message("py_env:" ${py_env})
H
Helin Wang 已提交
35
add_custom_command(OUTPUT ${OUTPUT_DIR}/.timestamp
L
liaogang 已提交
36
    COMMAND env ${py_env} ${PYTHON_EXECUTABLE} setup.py bdist_wheel
Z
zhangjinchao01 已提交
37
    COMMAND ${CMAKE_COMMAND} -E touch ${OUTPUT_DIR}/.timestamp
D
dongzhihong 已提交
38
    DEPENDS gen_proto_py ${PY_FILES} ${external_project_dependencies} ${COPY_PADDLE_MASTER})
Z
zhangjinchao01 已提交
39 40 41 42

add_custom_target(paddle_python ALL DEPENDS
    ${OUTPUT_DIR}/.timestamp)

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

45 46 47 48 49 50 51 52
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)
  endif()
53
endif()
Y
Yu Yang 已提交
54
install(DIRECTORY ${PADDLE_PYTHON_PACKAGE_DIR}
Z
zhangjinchao01 已提交
55 56
    DESTINATION opt/paddle/share/wheels
)