xla

8fc8d57a - Fix global_device_count(), local_device_count() for single process on CUDA (#6022)