xla

Set devices argument in dist.Backend.register_backend call.

#6254

Merged