update_test

Author: SHshenhao

.github/workflows/ci.yml

@@ -142,6 +142,7 @@ jobs:
             export HTTPS_PROXY=http://127.0.0.1:7897
             export SLIME_VISIBLE_DEVICES=mlx5_0,mlx5_1,mlx5_2,mlx5_3,mlx5_4,mlx5_5,mlx5_6,mlx5_7
             unset PIP_EXTRA_INDEX_URL
+            export DLSLIME_TCP_TEST_CUDA=ON
             python -m pip install -U pip pytest
             python -m pip show dlslime || true
             python -m pip uninstall -y dlslime || true

dlslime/bench/python/tcp_bench_spmd.py

@@ -120,11 +120,11 @@ def rank_0_print(*args):
     raise ValueError("Immediate data can only be used with write operations.")
 
 if args.transfer_engine == "dlslime":
-    tcp_endpoint = TcpEndpoint(f"{local_ip}", 12000 + local_rank)
+    tcp_endpoint = TcpEndpoint(f"{local_ip}", 22500 + local_rank)
 elif args.transfer_engine == "mooncake":
     engine = MooncakeTransferEngine()
     result = engine.initialize(
-        f"{local_ip}:{12000+local_rank}", "P2PHANDSHAKE", "tcp", None
+        f"{local_ip}:{22500+local_rank}", "P2PHANDSHAKE", "tcp", None
     )
     mooncake_endpoint_info = {
         "local_ip": local_ip,