[BUG] Fix bug for pd in fd (#3034)

* Fix bug for pd in fd * Fix bug for pd in fd --------- Co-authored-by: Jiang-Jia-Jun <163579578+Jiang-Jia-Jun@users.noreply.github.com>
2025-10-04 08:16:42 +08:00 · 2025-07-31 20:17:27 +08:00
parent 1a543bca29
commit fe17410f9c
2 changed files with 6 additions and 5 deletions
--- a/fastdeploy/cache_manager/transfer_factory/rdma_cache_transfer.py
+++ b/fastdeploy/cache_manager/transfer_factory/rdma_cache_transfer.py
@@ -45,7 +45,7 @@ class RDMACommManager:
            return
        self.messager = rdma_comm.RDMACommunicator(
            splitwise_role,
-            rank,
+            gpu_id,
            str(rdma_port) if splitwise_role == "decode" else "0",
            cache_k_ptr_list,
            cache_v_ptr_list,