Skip to content

Commit 2a63652

Browse files
authored
fix cmake error (#10729)
* fix cmake error * update fix
1 parent ca0dce7 commit 2a63652

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

paddle/fluid/operators/CMakeLists.txt

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -186,11 +186,7 @@ endif()
186186

187187
add_subdirectory(detail)
188188
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+
194190
set(DISTRIBUTE_DEPS sendrecvop_grpc grpc++_unsecure grpc_unsecure gpr cares zlib protobuf)
195191
set(DISTRIBUTE_COMPILE_FLAGS "-Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor")
196192
op_library(send_op DEPS ${DISTRIBUTE_DEPS})
@@ -208,6 +204,12 @@ if(WITH_DISTRIBUTE)
208204
set_source_files_properties(send_recv_op_test.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
209205
cc_test(test_send_recv SRCS send_recv_op_test.cc DEPS prefetch_op send_op listen_and_serv_op sum_op executor)
210206
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()
211213
else()
212214
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)
213215
endif()

0 commit comments

Comments
 (0)