xla

b1fbb749 - support dist.reduce_scatter_tensor (#7950)