Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
260 changes: 114 additions & 146 deletions paddle/fluid/distributed/fleet_executor/test/CMakeLists.txt
Original file line number Diff line number Diff line change
@@ -1,154 +1,122 @@
get_property(paddle_lib GLOBAL PROPERTY PADDLE_LIB_NAME)
set_source_files_properties(
interceptor_ping_pong_test.cc PROPERTIES COMPILE_FLAGS
${DISTRIBUTE_COMPILE_FLAGS})
if(WIN32 AND WITH_TESTING)
cc_test_old(interceptor_ping_pong_test SRCS interceptor_ping_pong_test.cc
DEPS fleet_executor ${BRPC_DEPS})
else()
cc_test_old(
interceptor_ping_pong_test
SRCS
interceptor_ping_pong_test.cc
DEPS
${paddle_lib}
python
fleet_executor)
endif()
# NOTE(risemeup1):The compile target of these unit tests requires occupying more than 4GB of memory,
# which lead a intractable compilation failure issue on Coverage CI.
# We temporarily disabled them, and will restore after all compilation issues are resolved.

set_source_files_properties(
compute_interceptor_test.cc PROPERTIES COMPILE_FLAGS
${DISTRIBUTE_COMPILE_FLAGS})
# get_property(paddle_lib GLOBAL PROPERTY PADDLE_LIB_NAME)
# set_source_files_properties(
# interceptor_ping_pong_test.cc PROPERTIES COMPILE_FLAGS
# ${DISTRIBUTE_COMPILE_FLAGS})
# if(WIN32 AND WITH_TESTING)
# cc_test_old(interceptor_ping_pong_test SRCS interceptor_ping_pong_test.cc
# DEPS fleet_executor ${BRPC_DEPS})
# else()
# cc_test_old(
# interceptor_ping_pong_test
# SRCS
# interceptor_ping_pong_test.cc
# DEPS
# ${paddle_lib}
# python
# fleet_executor)
# endif()

if(WIN32 AND WITH_TESTING)
cc_test_old(compute_interceptor_test SRCS compute_interceptor_test.cc DEPS
fleet_executor ${BRPC_DEPS})
else()
cc_test_old(
compute_interceptor_test
SRCS
compute_interceptor_test.cc
DEPS
${paddle_lib}
python
fleet_executor)
endif()
# set_source_files_properties(
# compute_interceptor_test.cc PROPERTIES COMPILE_FLAGS
# ${DISTRIBUTE_COMPILE_FLAGS})

set_source_files_properties(
source_interceptor_test.cc PROPERTIES COMPILE_FLAGS
${DISTRIBUTE_COMPILE_FLAGS})
if(WIN32 AND WITH_TESTING)
cc_test_old(source_interceptor_test SRCS source_interceptor_test.cc DEPS
fleet_executor ${BRPC_DEPS})
else()
cc_test_old(
source_interceptor_test
SRCS
source_interceptor_test.cc
DEPS
${paddle_lib}
python
fleet_executor)
endif()
# if(WIN32 AND WITH_TESTING)
# cc_test_old(compute_interceptor_test SRCS compute_interceptor_test.cc DEPS
# fleet_executor ${BRPC_DEPS})
# else()
# cc_test_old(compute_interceptor_test SRCS compute_interceptor_test.cc DEPS
# ${paddle_lib} python)
# endif()

set_source_files_properties(
sink_interceptor_test.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
if(WIN32 AND WITH_TESTING)
cc_test_old(sink_interceptor_test SRCS sink_interceptor_test.cc DEPS
fleet_executor ${BRPC_DEPS})
else()
cc_test_old(
sink_interceptor_test
SRCS
sink_interceptor_test.cc
DEPS
${paddle_lib}
python
fleet_executor)
endif()
# set_source_files_properties(
# source_interceptor_test.cc PROPERTIES COMPILE_FLAGS
# ${DISTRIBUTE_COMPILE_FLAGS})
# if(WIN32 AND WITH_TESTING)
# cc_test_old(source_interceptor_test SRCS source_interceptor_test.cc DEPS
# fleet_executor ${BRPC_DEPS})
# else()
# cc_test_old(source_interceptor_test SRCS source_interceptor_test.cc DEPS
# ${paddle_lib} python)
# endif()

set_source_files_properties(
interceptor_pipeline_short_path_test.cc
PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
if(WIN32 AND WITH_TESTING)
cc_test_old(
interceptor_pipeline_short_path_test SRCS
interceptor_pipeline_short_path_test.cc DEPS fleet_executor ${BRPC_DEPS})
else()
cc_test_old(
interceptor_pipeline_short_path_test
SRCS
interceptor_pipeline_short_path_test.cc
DEPS
${paddle_lib}
python
fleet_executor)
endif()
# set_source_files_properties(
# sink_interceptor_test.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
# if(WIN32 AND WITH_TESTING)
# cc_test_old(sink_interceptor_test SRCS sink_interceptor_test.cc DEPS
# fleet_executor ${BRPC_DEPS})
# else()
# cc_test_old(sink_interceptor_test SRCS sink_interceptor_test.cc DEPS
# ${paddle_lib} python)
# endif()

set_source_files_properties(
interceptor_pipeline_long_path_test.cc PROPERTIES COMPILE_FLAGS
${DISTRIBUTE_COMPILE_FLAGS})
if(WIN32 AND WITH_TESTING)
cc_test_old(
interceptor_pipeline_long_path_test SRCS
interceptor_pipeline_long_path_test.cc DEPS fleet_executor ${BRPC_DEPS})
else()
cc_test_old(
interceptor_pipeline_long_path_test
SRCS
interceptor_pipeline_long_path_test.cc
DEPS
${paddle_lib}
python
fleet_executor)
endif()
# set_source_files_properties(
# interceptor_pipeline_short_path_test.cc
# PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
# if(WIN32 AND WITH_TESTING)
# cc_test_old(
# interceptor_pipeline_short_path_test SRCS
# interceptor_pipeline_short_path_test.cc DEPS fleet_executor ${BRPC_DEPS})
# else()
# cc_test_old(interceptor_pipeline_short_path_test SRCS
# interceptor_pipeline_short_path_test.cc DEPS ${paddle_lib} python)
# endif()

set_source_files_properties(
compute_interceptor_run_op_test.cc PROPERTIES COMPILE_FLAGS
${DISTRIBUTE_COMPILE_FLAGS})
if(WIN32 AND WITH_TESTING)
cc_test_old(
compute_interceptor_run_op_test
SRCS
compute_interceptor_run_op_test.cc
DEPS
fleet_executor
naive_executor
fill_constant_op
op_registry
elementwise_add_op
scope
device_context
${BRPC_DEPS})
else()
cc_test_old(
compute_interceptor_run_op_test
SRCS
compute_interceptor_run_op_test.cc
DEPS
${paddle_lib}
python
fleet_executor
fill_constant_op
elementwise_add_op)
endif()
# set_source_files_properties(
# interceptor_pipeline_long_path_test.cc PROPERTIES COMPILE_FLAGS
# ${DISTRIBUTE_COMPILE_FLAGS})
# if(WIN32 AND WITH_TESTING)
# cc_test_old(
# interceptor_pipeline_long_path_test SRCS
# interceptor_pipeline_long_path_test.cc DEPS fleet_executor ${BRPC_DEPS})
# else()
# cc_test_old(interceptor_pipeline_long_path_test SRCS
# interceptor_pipeline_long_path_test.cc DEPS ${paddle_lib} python)
# endif()

if(WITH_DISTRIBUTE AND NOT WITH_PSLIB)
set_source_files_properties(
interceptor_ping_pong_with_brpc_test.cc
PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
if(WIN32 AND WITH_TESTING)
cc_test_old(
interceptor_ping_pong_with_brpc_test SRCS
interceptor_ping_pong_with_brpc_test.cc DEPS fleet_executor ${BRPC_DEPS})
else()
cc_test_old(
interceptor_ping_pong_with_brpc_test
SRCS
interceptor_ping_pong_with_brpc_test.cc
DEPS
${paddle_lib}
python
fleet_executor)
endif()
endif()
# set_source_files_properties(
# compute_interceptor_run_op_test.cc PROPERTIES COMPILE_FLAGS
# ${DISTRIBUTE_COMPILE_FLAGS})
# if(WIN32 AND WITH_TESTING)
# cc_test_old(
# compute_interceptor_run_op_test
# SRCS
# compute_interceptor_run_op_test.cc
# DEPS
# fleet_executor
# naive_executor
# fill_constant_op
# op_registry
# elementwise_add_op
# scope
# device_context
# ${BRPC_DEPS})
# else()
# cc_test_old(
# compute_interceptor_run_op_test
# SRCS
# compute_interceptor_run_op_test.cc
# DEPS
# ${paddle_lib}
# python
# fill_constant_op
# elementwise_add_op)
# endif()

# if(WITH_DISTRIBUTE AND NOT WITH_PSLIB)
# set_source_files_properties(
# interceptor_ping_pong_with_brpc_test.cc
# PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
# if(WIN32 AND WITH_TESTING)
# cc_test_old(
# interceptor_ping_pong_with_brpc_test SRCS
# interceptor_ping_pong_with_brpc_test.cc DEPS fleet_executor ${BRPC_DEPS})
# else()
# cc_test_old(
# interceptor_ping_pong_with_brpc_test SRCS
# interceptor_ping_pong_with_brpc_test.cc DEPS ${paddle_lib} python)
# endif()
# endif()