File tree Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -186,11 +186,7 @@ endif()
186
186
187
187
add_subdirectory (detail )
188
188
if (WITH_DISTRIBUTE )
189
- if (WITH_GPU )
190
- op_library (gen_nccl_id_op DEPS nccl_common )
191
- else ()
192
- set (DEPS_OPS ${DEPS_OPS} gen_nccl_id_op )
193
- endif ()
189
+
194
190
set (DISTRIBUTE_DEPS sendrecvop_grpc grpc++_unsecure grpc_unsecure gpr cares zlib protobuf )
195
191
set (DISTRIBUTE_COMPILE_FLAGS "-Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor" )
196
192
op_library (send_op DEPS ${DISTRIBUTE_DEPS} )
@@ -208,6 +204,12 @@ if(WITH_DISTRIBUTE)
208
204
set_source_files_properties (send_recv_op_test.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS} )
209
205
cc_test (test_send_recv SRCS send_recv_op_test.cc DEPS prefetch_op send_op listen_and_serv_op sum_op executor )
210
206
cc_test (test_send_nccl_id SRCS test_send_nccl_id.cc DEPS send_op listen_and_serv_op executor )
207
+ if (WITH_GPU )
208
+ op_library (gen_nccl_id_op DEPS nccl_common sendrecvop_grpc )
209
+ set_source_files_properties (gen_nccl_id_op.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS} )
210
+ else ()
211
+ set (DEPS_OPS ${DEPS_OPS} gen_nccl_id_op )
212
+ endif ()
211
213
else ()
212
214
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 )
213
215
endif ()
You can’t perform that action at this time.
0 commit comments