[MISC] Make GroupCoordinator compatible with out-of-tree devices (vllm-project#16464)

Signed-off-by: hzji210@gmail.com <hzji210@gmail.com>
Signed-off-by: Mu Huai <tianbowen.tbw@antgroup.com>

Author: ji-huazhong

vllm/distributed/parallel_state.py

@@ -194,9 +194,11 @@ def __init__(
 
         from vllm.platforms import current_platform
 
-        # TODO: fix it for other platforms
         if current_platform.is_cuda_alike():
             self.device = torch.device(f"cuda:{local_rank}")
+        elif current_platform.is_out_of_tree():
+            self.device = torch.device(
+                f"{current_platform.device_name}:{local_rank}")
         else:
             self.device = torch.device("cpu")