openxla / xla

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

Apache License 2.0

2.71k stars 436 forks source link

#sdy Refactor `xla-sdy-mhlo-round-trip-shard-map-export` from a `ConversionPattern` to a walk. #19658

Closed copybara-service[bot] closed 7 hours ago

copybara-service[bot] commented 1 day ago

sdy Refactor `xla-sdy-mhlo-round-trip-shard-map-export` from a `ConversionPattern` to a walk.