xla

f5f27268 - Add out-of-place reduce-scatter coalescing (#6058)