xla

5aa1ac3d - [PJRT] Cache world size and ordinal only when addressable_device_count == 1 (#4965)