CMakeLists.txt 608 字节
Newer Older
1
cc_library(processgroup SRCS ProcessGroup.cc DEPS phi phi_api eager_api)
2 3 4
if (WITH_DISTRIBUTE)
  cc_library(processgroup_gloo SRCS ProcessGroupGloo.cc DEPS phi phi_api eager_api gloo_wrapper)
endif()
5
cc_library(eager_reducer SRCS reducer.cc DEPS eager_api processgroup)
6 7

if(WITH_NCCL)
8
    cc_library(processgroup_nccl SRCS ProcessGroupNCCL.cc DEPS place cuda_stream enforce collective_helper device_context phi phi_api eager_api)
9
endif()
10 11 12
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()