CMakeLists.txt 1.3 KB
Newer Older
1
cc_library(
W
Wen Sun 已提交
2
  process_group
W
Wen Sun 已提交
3
  SRCS process_group.cc
L
LiYuRio 已提交
4
  DEPS dense_tensor)
5
cc_library(
W
Wen Sun 已提交
6 7
  process_group_stream
  SRCS process_group_stream.cc
8
  DEPS dense_tensor)
9 10 11
cc_library(
  eager_reducer
  SRCS reducer.cc
W
Wen Sun 已提交
12
  DEPS eager_api process_group process_group_stream phi_api string_helper)
13

14 15
if(WITH_DISTRIBUTE)
  cc_library(
W
Wen Sun 已提交
16 17
    process_group_gloo
    SRCS process_group_gloo.cc
18
    DEPS phi_api eager_api gloo_wrapper)
19
endif()
20

S
ShenLiang 已提交
21
if(WITH_NCCL OR WITH_RCCL)
22
  cc_library(
W
Wen Sun 已提交
23
    process_group_nccl
W
Wen Sun 已提交
24 25
    SRCS process_group_nccl.cc nccl_tools.cc common.cc check.cc
    DEPS process_group
W
Wen Sun 已提交
26
         process_group_stream
27 28 29 30
         place
         enforce
         collective_helper
         device_context
31
         ${DEVICE_EVENT_LIBS}
L
LiYuRio 已提交
32
         dense_tensor)
33
endif()
L
lilong12 已提交
34

J
james 已提交
35 36
if(WITH_XPU_BKCL)
  cc_library(
W
Wen Sun 已提交
37
    process_group_bkcl
W
Wen Sun 已提交
38
    SRCS process_group_bkcl.cc bkcl_tools.cc common.cc
W
Wen Sun 已提交
39
    DEPS process_group place enforce collective_helper device_context
J
james 已提交
40 41 42
         dense_tensor)
endif()

W
wuhuachaocoding 已提交
43 44
if(WITH_MPI)
  cc_library(
W
Wen Sun 已提交
45 46
    process_group_mpi
    SRCS process_group_mpi.cc mpi_tools.cc common.cc
W
wuhuachaocoding 已提交
47 48 49
    DEPS collective_helper device_context)
endif()

50 51
if(WITH_CUSTOM_DEVICE)
  cc_library(
W
Wen Sun 已提交
52
    process_group_custom
W
Wen Sun 已提交
53
    SRCS process_group_custom.cc custom_ccl_tools.cc common.cc
W
Wen Sun 已提交
54
    DEPS process_group phi_backends place enforce collective_helper
55
         device_context)
56
endif()