openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
Apache License 2.0
2.55k stars 394 forks source link

[XLA:SPMD] Add a `MergeShardingIfCompatible` overload that doesn't require minimum_tiles. #14883

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

[XLA:SPMD] Add a MergeShardingIfCompatible overload that doesn't require minimum_tiles.

This is a small cleanup that makes the code more concise. There is no behavior change.