|
|
|
@ -204,6 +204,7 @@ 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)
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
set_source_files_properties(test_send_nccl_id.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
|
|
|
|
|
cc_test(test_send_nccl_id SRCS test_send_nccl_id.cc DEPS send_op listen_and_serv_op executor)
|
|
|
|
|
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})
|
|
|
|
|