Add CI to test args -a deepep

.github/workflows/pr-test-npu.yml

@@ -17,7 +17,7 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  test-low-latency:
+  test-all-build:
     if: (github.repository == 'sgl-project/sgl-kernel-npu' || github.event_name == 'pull_request') &&
         github.event.pull_request.draft == false
     runs-on: linux-aarch64-a3-16
@@ -52,13 +52,14 @@ jobs:
         run: bash scripts/prepare_deepep_in_container.sh
 
       - name: Run test
-        timeout-minutes: 10
+        timeout-minutes: 20
         env:
-          HCCL_BUFFSIZE: 2000
+          HCCL_BUFFSIZE: 3000
         run: |
+          python3 $GITHUB_WORKSPACE/tests/python/deepep/test_intranode.py
           python3 $GITHUB_WORKSPACE/tests/python/deepep/test_low_latency.py
 
-  test-intranode:
+  test-build-deepep:
     if: (github.repository == 'sgl-project/sgl-kernel-npu' || github.event_name == 'pull_request') &&
         github.event.pull_request.draft == false
     runs-on: linux-aarch64-a3-16
@@ -90,20 +91,21 @@ jobs:
           bash scripts/npu_ci_install_dependency.sh
 
       - name: Prepare Deepep
-        run: bash scripts/prepare_deepep_in_container.sh
+        run: bash scripts/prepare_deepep_in_container.sh -a deepep
 
       - name: Run test
-        timeout-minutes: 10
+        timeout-minutes: 20
         env:
           HCCL_BUFFSIZE: 3000
         run: |
           python3 $GITHUB_WORKSPACE/tests/python/deepep/test_intranode.py
+          python3 $GITHUB_WORKSPACE/tests/python/deepep/test_low_latency.py
 
   finish:
     if: always()
     needs:
-      - test-low-latency
-      - test-intranode
+      - test-all-build
+      - test-build-deepep
     runs-on: ubuntu-latest
     steps:
       - name: Check all dependent job statuses

scripts/prepare_deepep_in_container.sh

@@ -1,5 +1,29 @@
+#!/bin/bash
+
+while getopts ":a:" opt; do
+    case ${opt} in
+        a )
+            BUILD_ARGS="$OPTARG"
+            ;;
+        \? )
+            echo "Error: unknown flag: -$OPTARG" 1>&2
+            exit 1
+            ;;
+        : )
+            echo "Error: -$OPTARG requires a value" 1>&2
+            exit 1
+            ;;
+    esac
+done
+
+shift $((OPTIND -1))
+
 cd ${GITHUB_WORKSPACE}
-bash build.sh
+if [ -n "$BUILD_ARGS" ]; then
+    bash build.sh -a "$BUILD_ARGS"
+else
+    bash build.sh
+fi
 pip install ${GITHUB_WORKSPACE}/output/deep_ep*.whl --no-cache-dir
 cd "$(pip show deep-ep | awk '/^Location:/ {print $2}')"
 ln -s deep_ep/deep_ep_cpp*.so || true