add flagcx version select logic

Author: leoda1

vllm_fl/distributed/device_communicators/flagcx.py

@@ -96,9 +96,20 @@ def __init__(
         self.available = True
         self.disabled = False
 
+        try:
+            _flagcx_version = ctypes.c_int()
+            self.flagcx._funcs["flagcxGetVersion"](
+                ctypes.byref(_flagcx_version))
+            self._legacy_unique_id_api = _flagcx_version.value < 1300
+        except (AttributeError, KeyError, TypeError):
+            self._legacy_unique_id_api = False
+
         if self.rank == 0:
             # get the unique id from NCCL
-            self.unique_id = self.flagcx.flagcxGetUniqueId()
+            if self._legacy_unique_id_api:
+                self.unique_id = self.flagcx.flagcxGetUniqueId().contents
+            else:
+                self.unique_id = self.flagcx.flagcxGetUniqueId()
         else:
             # construct an empty unique id
             self.unique_id = flagcxUniqueId()
@@ -133,8 +144,12 @@ def __init__(
             device_ctx = torch.cuda.device(self.device)
 
         with device_ctx:
-            self.comm = self.flagcx.flagcxCommInitRank(
-                self.world_size, self.unique_id, self.rank)
+            if self._legacy_unique_id_api:
+                self.comm = self.flagcx.flagcxCommInitRank(
+                    self.world_size, ctypes.byref(self.unique_id), self.rank)
+            else:
+                self.comm = self.flagcx.flagcxCommInitRank(
+                    self.world_size, self.unique_id, self.rank)
 
             stream = current_stream()
             # A small all_reduce for warmup.