xla
[SPMD] Fix reduce_scatter for TPUv2
#7248
Merged

Loading