search

openxla / xla

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

Use `if_google` instead of buildozer transformations in `auto_sharding` #14248

Closed copybara-service[bot] closed 3 months ago

copybara-service[bot] commented 3 months ago

Use if_google instead of buildozer transformations in auto_sharding

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/14103 from shraiysh:enable_send_recv_validation 3a9628713c49d0966fae4fb15484762e19133435