xla
Fix global_runtime_device_count() for spmd case
#6266
Merged

Loading