upgrade torch version for sgl-kernel (#3026)

This commit is contained in:
Yineng Zhang
2025-01-21 14:32:08 +08:00
committed by GitHub
parent 6c856b4f3a
commit ec1c21cdc4
2 changed files with 9 additions and 9 deletions

View File

@@ -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