[xla:cpu] InProcessCommunicator: compute collective operations in parallel using all ranks
PiperOrigin-RevId: 730502259
Showing
- third_party/xla/xla/backends/cpu/collectives/BUILD 4 additions, 0 deletionsthird_party/xla/xla/backends/cpu/collectives/BUILD
- third_party/xla/xla/backends/cpu/collectives/in_process_communicator.cc 162 additions, 94 deletions...a/xla/backends/cpu/collectives/in_process_communicator.cc
- third_party/xla/xla/service/rendezvous.h 19 additions, 4 deletionsthird_party/xla/xla/service/rendezvous.h
Please register or sign in to comment