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

4 5 6 7 8 9 10 11
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)

set(PY_FILES paddle/__init__.py
             ${TRAINER_PY_FILES}
             ${HELPERS_PY_FILES}
             ${UTILS_PY_FILES})
Z
zhangjinchao01 已提交
12 13 14 15 16

configure_file(${CMAKE_CURRENT_SOURCE_DIR}/setup.py.in
    ${CMAKE_CURRENT_BINARY_DIR}/setup.py)

add_custom_command(OUTPUT ${OUTPUT_DIR}/.timestamp
L
liaogang 已提交
17
    COMMAND env ${py_env} ${PYTHON_EXECUTABLE} setup.py bdist_wheel
Z
zhangjinchao01 已提交
18
    COMMAND ${CMAKE_COMMAND} -E touch ${OUTPUT_DIR}/.timestamp
L
liaogang 已提交
19
    DEPENDS gen_proto_py ${PY_FILES} ${external_project_dependencies})
Z
zhangjinchao01 已提交
20 21 22 23

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

24 25
add_subdirectory(paddle/trainer_config_helpers/tests)

Z
zhangjinchao01 已提交
26 27 28
install(DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/dist/
    DESTINATION opt/paddle/share/wheels
)