xla
f38248bb
- [SPMD] Set replicated output false (#5139)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Minimap (CTRL+M)
Commit
2 years ago
[SPMD] Set replicated output false (#5139) * [SPMD] update partial replication test * [SPMD] output sharding by default
References
#5139 - [SPMD] Set replicated output false
Author
yeounoh
Parents
02c41cbb
Files
2
test/spmd
test_xla_sharding.py
torch_xla/csrc
xla_sharding_util.h
Loading
xla_sharding_util.h