[NPU]Add Ascend transfer version compatibility. (#31189)

2026-07-26 21:07:59 +08:00
parent e8a635a412
commit e14068d161
1 changed files with 5 additions and 3 deletions
@@ -48,10 +48,12 @@ class AscendTransferEngine(MooncakeTransferEngine):
        else:
            logger.error(f"Unsupported DisaggregationMode: {disaggregation_mode}")
            raise ValueError(f"Unsupported DisaggregationMode: {disaggregation_mode}")
-        self.session_id = NetworkAddress(
-            self.hostname, self.engine.get_rpc_port()
-        ).to_host_port_str()
+        rpc_port = self.engine.get_rpc_port()
+        self.session_id = NetworkAddress(self.hostname, rpc_port).to_host_port_str()
        self.initialize()
+        if rpc_port == 0:
+            rpc_port = self.engine.get_rpc_port()
+            self.session_id = NetworkAddress(self.hostname, rpc_port).to_host_port_str()

    def initialize(self) -> None:
        from sglang.srt.distributed.parallel_state import (