Skip to content

Commit

Permalink
[xla:cpu] Move GlooCommunicator to backends/cpu/collectives
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 713129065
  • Loading branch information
ezhulenev authored and Google-ML-Automation committed Jan 8, 2025
1 parent f93bc74 commit 9d1dbbb
Show file tree
Hide file tree
Showing 6 changed files with 600 additions and 480 deletions.
41 changes: 41 additions & 0 deletions xla/backends/cpu/collectives/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,47 @@ cc_library(
],
)

# TODO(b/380457503): Restrict visibility to private.
cc_library(
name = "gloo_communicator",
srcs = ["gloo_communicator.cc"],
hdrs = ["gloo_communicator.h"],
copts = [
"-fexceptions",
"-fno-strict-aliasing",
],
features = ["-use_header_modules"],
deps = [
":cpu_collectives",
"//xla:shape_util",
"//xla:status_macros",
"//xla:types",
"//xla:util",
"//xla:xla_data_proto_cc",
"//xla/core/collectives:communicator",
"//xla/core/collectives:rank_id",
"//xla/service:collective_ops_utils",
"//xla/service:global_device_id",
"//xla/service/cpu:collectives_interface",
"//xla/stream_executor:device_memory",
"//xla/tsl/platform:errors",
"//xla/tsl/platform:statusor",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/log",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/synchronization",
"@com_google_absl//absl/time",
"@com_google_absl//absl/types:span",
"@gloo",
"@tsl//tsl/platform:errors",
"@tsl//tsl/platform:logging",
],
)

# TODO(b/380457503): Restrict visibility to private.
cc_library(
name = "in_process_communicator",
srcs = ["in_process_communicator.cc"],
Expand Down
Loading

0 comments on commit 9d1dbbb

Please sign in to comment.