[XLA:GPU] Support cross-replica cps in collective-permute decomposer - Githubissues

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

Apache License 2.0

2.72k stars 438 forks source link

[XLA:GPU] Support cross-replica cps in collective-permute decomposer #19676

Open copybara-service[bot] opened 4 days ago

copybara-service[bot] commented 4 days ago

[XLA:GPU] Support cross-replica cps in collective-permute decomposer