CMakeLists.txt 1.7 KB
Newer Older
1
add_subdirectory(collective)
2
add_subdirectory(store)
W
wangguanqun 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
if(WITH_PYTHON)
  py_proto_compile(ps_py_proto SRCS the_one_ps.proto)
  add_custom_target(ps_py_proto_init ALL  
    COMMAND ${CMAKE_COMMAND} -E make_directory ${PADDLE_BINARY_DIR}/python/paddle/distributed/fleet/proto)
  add_dependencies(ps_py_proto ps_py_proto_init)
  if (NOT WIN32)
    add_custom_command(TARGET ps_py_proto POST_BUILD
      COMMAND mv the_one_ps_pb2.py ${PADDLE_BINARY_DIR}/python/paddle/distributed/fleet/proto/)
  else(NOT WIN32)
    string(REPLACE "/" "\\" fleet_proto_dstpath "${PADDLE_BINARY_DIR}/python/paddle/distributed/fleet/proto/")
    add_custom_command(TARGET ps_py_proto POST_BUILD
      COMMAND copy /Y the_one_ps_pb2.py ${fleet_proto_dstpath}
      COMMENT "Copy generated python the_one_ps_pb2 into directory ${fleet_proto_dstpath}.")
  endif(NOT WIN32)
endif()

T
tangwei12 已提交
19
if(NOT WITH_PSCORE)
L
LiYuRio 已提交
20
    add_subdirectory(fleet_executor)
T
tangwei12 已提交
21 22 23
    return()
endif()

W
wangguanqun 已提交
24 25 26 27
proto_library(ps_framework_proto SRCS the_one_ps.proto)
add_custom_command(TARGET ps_framework_proto POST_BUILD
    COMMAND mv the_one_ps.pb.h ps.pb.h
    COMMAND mv the_one_ps.pb.cc ps.pb.cc)
T
tangwei12 已提交
28 29 30 31 32 33 34 35

set(DISTRIBUTE_COMPILE_FLAGS "-Wno-error=unused-value -Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor -Wno-error=sign-compare -Wno-error=unused-variable -Wno-error=return-type -Wno-error=unused-but-set-variable -Wno-error=type-limits -Wno-error=unknown-pragmas -Wno-error=parentheses -Wno-error=unused-result")

if (CMAKE_CXX_COMPILER_VERSION VERSION_GREATER 7.0)
    set(DISTRIBUTE_COMPILE_FLAGS
            "${DISTRIBUTE_COMPILE_FLAGS} -faligned-new")
endif()

36
add_subdirectory(common)
37
add_subdirectory(ps)
T
tangwei12 已提交
38
add_subdirectory(test)
1
123malin 已提交
39
add_subdirectory(index_dataset)
L
LiYuRio 已提交
40
add_subdirectory(fleet_executor)