|
|
|
@ -186,11 +186,7 @@ endif()
|
|
|
|
|
|
|
|
|
|
add_subdirectory(detail)
|
|
|
|
|
if(WITH_DISTRIBUTE)
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
op_library(gen_nccl_id_op DEPS nccl_common)
|
|
|
|
|
else()
|
|
|
|
|
set(DEPS_OPS ${DEPS_OPS} gen_nccl_id_op)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
set(DISTRIBUTE_DEPS sendrecvop_grpc grpc++_unsecure grpc_unsecure gpr cares zlib protobuf)
|
|
|
|
|
set(DISTRIBUTE_COMPILE_FLAGS "-Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor")
|
|
|
|
|
op_library(send_op DEPS ${DISTRIBUTE_DEPS})
|
|
|
|
@ -208,6 +204,12 @@ if(WITH_DISTRIBUTE)
|
|
|
|
|
set_source_files_properties(send_recv_op_test.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
|
|
|
|
|
cc_test(test_send_recv SRCS send_recv_op_test.cc DEPS prefetch_op send_op listen_and_serv_op sum_op executor)
|
|
|
|
|
cc_test(test_send_nccl_id SRCS test_send_nccl_id.cc DEPS send_op listen_and_serv_op executor)
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
op_library(gen_nccl_id_op DEPS nccl_common sendrecvop_grpc)
|
|
|
|
|
set_source_files_properties(gen_nccl_id_op.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
|
|
|
|
|
else()
|
|
|
|
|
set(DEPS_OPS ${DEPS_OPS} gen_nccl_id_op)
|
|
|
|
|
endif()
|
|
|
|
|
else()
|
|
|
|
|
set(DEPS_OPS ${DEPS_OPS} send_op prefetch_op recv_op listen_and_serv_op send_vars_op send_barrier_op gen_nccl_id_op)
|
|
|
|
|
endif()
|
|
|
|
|