diff --git a/.github/workflows/pr-test-npu.yml b/.github/workflows/pr-test-npu.yml index b9e37a29e..4395f7a2c 100644 --- a/.github/workflows/pr-test-npu.yml +++ b/.github/workflows/pr-test-npu.yml @@ -39,6 +39,9 @@ jobs: - name: Start CI container run: bash scripts/npu_ci_a3_start_container.sh + - name: Prepare Deepep + run: bash scripts/prepare_deepep_in_container.sh + - name: Run test timeout-minutes: 10 run: | @@ -69,6 +72,9 @@ jobs: - name: Start CI container run: bash scripts/npu_ci_a3_start_container.sh + - name: Prepare Deepep + run: bash scripts/prepare_deepep_in_container.sh + - name: Run test timeout-minutes: 10 run: | diff --git a/scripts/prepare_deepep_in_container.sh b/scripts/prepare_deepep_in_container.sh new file mode 100644 index 000000000..190055d14 --- /dev/null +++ b/scripts/prepare_deepep_in_container.sh @@ -0,0 +1,4 @@ +ENV_DIR="/home/runner/.cache/env_sglang.sh" +docker exec \ + sglang_kernel_ci_a3 \ + /bin/bash -c "source \"$ENV_DIR\" && cd ${GITHUB_WORKSPACE} && bash build.sh && 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