xla icon indicating copy to clipboard operation
xla copied to clipboard

[XLA:CollectivePipeliner] Fix two issues:

Open copybara-service[bot] opened this issue 1 week ago • 0 comments

[XLA:CollectivePipeliner] Fix two issues:

  1. Accept transpose as a formatting op in ForwardSink.
  2. Do not stop when a large collective was sunk in the previous iteration. Instead, delay sinking large collectives while sinking small collectives level by level and run an additional sinking iteration dedicated to large collectives at the end.

copybara-service[bot] avatar Nov 14 '25 22:11 copybara-service[bot]