moidfy to setup sglang

.azure-pipelines/templates/sglang-test.yaml

@@ -12,17 +12,19 @@ parameters:
   type: string
 
 steps:
-- task: Bash@3
-  name: Build
-  displayName: Build
-  inputs:
-    targetType: inline
-    script: |
-      git clone https://github.com/microsoft/mscclpp.git && cd mscclpp
-      mkdir build && cd build
-      cmake -DCMAKE_BUILD_TYPE=Release -DMSCCLPP_BYPASS_GPU_CHECK=ON -DMSCCLPP_USE_CUDA=ON -DMSCCLPP_BUILD_TESTS=ON -DMSCCLPP_GPU_ARCHS=${{ parameters.gpuArch }} ..
-      make -j
-    workingDirectory: '$(System.DefaultWorkingDirectory)'
+# - task: Bash@3
+#   name: Build
+#   displayName: Build
+#   inputs:
+#     targetType: inline
+#     script: |
+#       git clone https://github.com/microsoft/mscclpp.git && cd mscclpp
+#       pip install .
+#       pip install -r ./python/requirements_cuda13.txt
+#       mkdir build && cd build
+#       cmake -DCMAKE_BUILD_TYPE=Release -DMSCCLPP_BYPASS_GPU_CHECK=ON -DMSCCLPP_USE_CUDA=ON -DMSCCLPP_BUILD_TESTS=ON -DMSCCLPP_GPU_ARCHS=${{ parameters.gpuArch }} ..
+#       make -j
+#     workingDirectory: '$(System.DefaultWorkingDirectory)'
 
 - task: Bash@3
   name: InstallPackages
@@ -59,6 +61,35 @@ steps:
     arguments: "single-node-test"
     workingDirectory: '$(System.DefaultWorkingDirectory)'
 
+- task: Bash@3
+  name: SGLangSetup
+  displayName: SGLang Setup
+  inputs:
+    targetType: inline
+    script: |
+      set -e
+      HOSTFILE=$(System.DefaultWorkingDirectory)/test/deploy/hostfile_ci
+      SSH_OPTION="StrictHostKeyChecking=no"
+      KeyFilePath=${SSHKEYFILE_SECUREFILEPATH}
+      : > azureuser@10.0.0.4
+      tail -f azureuser@10.0.0.4 &
+      CHILD_PID=$!
+      parallel-ssh -o . -t 0 -h ${HOSTFILE} -x "-i ${KeyFilePath}" \
+        -O $SSH_OPTION 'sudo docker run -itd --name=mscclpp-test --privileged --net=host --ipc=host --gpus=all -w /root -v /mnt:/mnt lmsysorg/sglang:latest bash && \
+        sudo docker exec -t mscclpp-test bash -c " \
+          git clone https://github.com/microsoft/mscclpp.git && \
+          cd mscclpp && \
+          mkdir build && \
+          cd build && \
+          cmake -DCMAKE_BUILD_TYPE=Release .. && \
+          make -j && \
+          cd .. && \
+          pip install . && \
+          pip install -r ./python/requirements_cuda13.txt \
+        "'
+      kill $CHILD_PID
+    workingDirectory: '$(System.DefaultWorkingDirectory)'
+
 - task: Bash@3
   name: AllGatherTest
   displayName: Run mscclpp AllGather test