|
|
@ -33,7 +33,7 @@ register_operators(EXCLUDES gen_nccl_id_op DEPS ${DISTRIBUTE_DEPS})
|
|
|
|
|
|
|
|
|
|
|
|
if(WITH_GPU AND NOT WIN32)
|
|
|
|
if(WITH_GPU AND NOT WIN32)
|
|
|
|
set(DISTRIBUTE_DEPS ${DISTRIBUTE_DEPS} nccl_common)
|
|
|
|
set(DISTRIBUTE_DEPS ${DISTRIBUTE_DEPS} nccl_common)
|
|
|
|
op_library(gen_nccl_id_op ${DISTRIBUTE_DEPS} nccl_common)
|
|
|
|
op_library(gen_nccl_id_op DEPS ${DISTRIBUTE_DEPS} nccl_common)
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
|
|
set(OPERATOR_DEPS ${OPERATOR_DEPS} ${DISTRIBUTE_DEPS} PARENT_SCOPE)
|
|
|
|
set(OPERATOR_DEPS ${OPERATOR_DEPS} ${DISTRIBUTE_DEPS} PARENT_SCOPE)
|
|
|
|