xla
0025ca76
- [SPMD] Fix reduce_scatter for TPUv2 (#7248)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[SPMD] Fix reduce_scatter for TPUv2 (#7248) Summary: Fix reduce_scatter test for TPUv2 Test Plan: python test/spmd/test_xla_sharding.py -v -k test_spmd_reduce_scatter
References
#7248 - [SPMD] Fix reduce_scatter for TPUv2
Author
alanwaketan
Parents
c216d26c
Loading