xla

c8a725c9 - Fix `global_device_map` sorting (#2630)