gemini suggestion

Signed-off-by: Lehui Liu <[email protected]>

Author: liulehui

python/ray/train/torch/config.py

@@ -149,7 +149,7 @@ def _setup_torch_process_group(
             return
         print(">>> os.environ[\"XLA_USE_SPMD\"] = ", os.environ["XLA_USE_SPMD"])
         dist.init_process_group(
-            backend='gloo',
+            backend='xla',
             init_method='env://',
             rank=world_rank,
             world_size=world_size,