CMakeLists.txt 481 字节
Newer Older
1 2
proto_library(auto_parallel_proto SRCS auto_parallel.proto)

3 4 5 6 7
cc_library(
  device_mesh
  SRCS device_mesh.cc
  DEPS auto_parallel_proto)

8 9 10 11
cc_library(
  process_mesh
  SRCS process_mesh.cc
  DEPS auto_parallel_proto)
12

13 14 15 16 17
cc_library(
  dist_attr
  SRCS dist_attr.cc
  DEPS process_mesh auto_parallel_proto proto_desc)

18 19 20 21
cc_library(
  dist_mapper
  SRCS dist_mapper.cc
  DEPS device_mesh auto_parallel_proto)
22

23
cc_library(auto_parallel DEPS device_mesh process_mesh dist_attr dist_mapper)