CMakeLists.txt 621 字节
Newer Older
1
cc_library(processgroup SRCS ProcessGroup.cc DEPS phi phi_api eager_api)
2 3
cc_library(eager_reducer SRCS reducer.cc DEPS eager_api processgroup phi phi_api)

4 5 6
if (WITH_DISTRIBUTE)
  cc_library(processgroup_gloo SRCS ProcessGroupGloo.cc DEPS phi phi_api eager_api gloo_wrapper)
endif()
7 8

if(WITH_NCCL)
9
    cc_library(processgroup_nccl SRCS ProcessGroupNCCL.cc DEPS place cuda_stream enforce collective_helper device_context phi phi_api eager_api)
10
endif()
11 12 13
if(WITH_ASCEND_CL)
    cc_library(processgroup_hccl SRCS ProcessGroupHCCL.cc DEPS place npu_stream enforce collective_helper device_context phi phi_api eager_api)
endif()