Tcp End

.github/workflows/ci.yml

@@ -55,7 +55,8 @@ jobs:
             cmake \
             ninja-build \
             libnuma-dev \
-            libibverbs-dev
+            libibverbs-dev \
+            libasio-dev
 
       - name: Install build dependencies
         run: |
@@ -66,14 +67,17 @@ jobs:
         env:
           CMAKE_ARGS: >-
             -DBUILD_RDMA=ON
+            -DBUILD_TCP=ON
             -DBUILD_PYTHON=ON
             -DBUILD_NVLINK=OFF
             -DBUILD_TORCH_PLUGIN=OFF
             -DBUILD_ASCEND_DIRECT=OFF
             -DBUILD_TEST=OFF
+            -DUSE_CUDA=OFF
         # The native wheel lives under dlslime/ (dlslime-ctrl/ is a separate Rust crate).
         run: python -m build --wheel --outdir dist dlslime
 
+
       - name: Install wheel smoke test
         run: |
           python -m pip install dist/dlslime-*.whl --no-deps
@@ -128,6 +132,8 @@ jobs:
 
           docker exec "${container_name}" bash -lc '
             set -euxo pipefail
+            apt install -y libasio-dev
+
             cd /workspace
             export PIP_CONFIG_FILE=/dev/null
             export PIP_INDEX_URL=https://mirrors.aliyun.com/pypi/simple/
@@ -138,6 +144,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/CMakeLists.txt

@@ -13,6 +13,7 @@ slime_option(USE_MACA                  "USE in MACA Platform"             OFF)
 
 slime_option(BUILD_NVLINK              "Build NVLINK"                     OFF)
 slime_option(BUILD_ASCEND_DIRECT       "Build Ascend direct transport"    OFF)
+slime_option(BUILD_TCP                 "Build TCP transport"               ON)
 
 # Slime options for custom python wrapper
 slime_option(BUILD_PYTHON              "Build python wrapper"             OFF)