|
|
|
@ -54,7 +54,7 @@ cc_test(varhandle_test SRCS varhandle_test.cc DEPS profiler scope)
|
|
|
|
|
cc_library(parameter_prefetch SRCS parameter_prefetch.cc DEPS sendrecvop_rpc memory)
|
|
|
|
|
cc_library(parameter_send SRCS parameter_send.cc DEPS sendrecvop_rpc memory)
|
|
|
|
|
cc_library(parameter_recv SRCS parameter_recv.cc DEPS sendrecvop_rpc memory)
|
|
|
|
|
cc_library(communicator SRCS communicator.cc DEPS scope selected_rows tensor variable_helper selected_rows_functor simple_threadpool)
|
|
|
|
|
cc_library(communicator SRCS communicator.cc DEPS scope selected_rows tensor variable_helper selected_rows_functor simple_threadpool parameter_send parameter_recv)
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
cc_test(collective_server_test SRCS collective_server_test.cc
|
|
|
|
|
DEPS sendrecvop_rpc executor ${RPC_DEPS}
|
|
|
|
|