xla
5eaea4c8
- [SPMD] Use d2d replication via CopyToDevice (#4336)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Minimap (CTRL+M)
Commit
2 years ago
[SPMD] Use d2d replication via CopyToDevice (#4336)
References
#4336 - [SPMD] Use d2d replication via CopyToDevice
Author
yeounoh
Parents
3f015284
Files
5
third_party/xla_client
computation_client.h
pjrt_computation_client.cc
pjrt_computation_client.h
xrt_computation_client.h
torch_xla/csrc
xla_sharding_util.cpp
Loading
pjrt_computation_client.ccpjrt_computation_client.hxrt_computation_client.hxla_sharding_util.cpp