xla
[SPMD] Support reduce-scatter in manual sharding
#7231
Merged

Loading