diff --git a/.github/workflows/pr-test-sgl-kernel.yml b/.github/workflows/pr-test-sgl-kernel.yml index cacf938a3..31360c0a0 100644 --- a/.github/workflows/pr-test-sgl-kernel.yml +++ b/.github/workflows/pr-test-sgl-kernel.yml @@ -34,16 +34,16 @@ jobs: if: github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request' runs-on: 1-gpu-runner steps: - - name: Checkout code - uses: actions/checkout@v3 + - uses: actions/checkout@v4 + with: + submodules: 'recursive' - - name: Install dependencies + - name: Install run: | - bash scripts/ci_install_dependency.sh - + pip3 install torch==2.5.1 + pip3 uninstall sgl-kernel -y || true cd sgl-kernel - git submodule update --init --recursive - pip3 install -e . --force-reinstall + pip3 install . pip3 list | grep sgl-kernel - name: Run test @@ -57,7 +57,7 @@ jobs: pip3 uninstall sgl-kernel -y finish: - needs: [unit-test] + needs: [unit-test, lint] runs-on: ubuntu-latest steps: - name: Finish diff --git a/sgl-kernel/build.sh b/sgl-kernel/build.sh index 55ce9df7f..0d8169579 100755 --- a/sgl-kernel/build.sh +++ b/sgl-kernel/build.sh @@ -8,7 +8,7 @@ docker run --rm \ -v "$(pwd)":/sgl-kernel \ pytorch/manylinux-builder:cuda${CUDA_VERSION} \ bash -c " - ${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir torch==2.4.0 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.} && \ + ${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir torch==2.5.1 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.} && \ export TORCH_CUDA_ARCH_LIST='7.5 8.0 8.9 9.0+PTX' && \ export CUDA_VERSION=${CUDA_VERSION} && \ mkdir -p /usr/lib/x86_64-linux-gnu/ && \