openxla / xla

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

If the auto-sharding pass cannot find a solution for any of the mesh shapes tried, return an error message listing the errors encountered for each of the mesh shapes. This is intended to make debugging easier. #17901

Closed copybara-service[bot] closed 1 week ago

copybara-service[bot] commented 1 week ago

If the auto-sharding pass cannot find a solution for any of the mesh shapes tried, return an error message listing the errors encountered for each of the mesh shapes. This is intended to make debugging easier.

Reverts 410db7ba3541f4b87911e96555d2eb4531465e9a