Unverified Commit ec1c21cd authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

upgrade torch version for sgl-kernel (#3026)

parent 6c856b4f
...@@ -34,16 +34,16 @@ jobs: ...@@ -34,16 +34,16 @@ jobs:
if: github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request' if: github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request'
runs-on: 1-gpu-runner runs-on: 1-gpu-runner
steps: steps:
- name: Checkout code - uses: actions/checkout@v4
uses: actions/checkout@v3 with:
submodules: 'recursive'
- name: Install dependencies - name: Install
run: | run: |
bash scripts/ci_install_dependency.sh pip3 install torch==2.5.1
pip3 uninstall sgl-kernel -y || true
cd sgl-kernel cd sgl-kernel
git submodule update --init --recursive pip3 install .
pip3 install -e . --force-reinstall
pip3 list | grep sgl-kernel pip3 list | grep sgl-kernel
- name: Run test - name: Run test
...@@ -57,7 +57,7 @@ jobs: ...@@ -57,7 +57,7 @@ jobs:
pip3 uninstall sgl-kernel -y pip3 uninstall sgl-kernel -y
finish: finish:
needs: [unit-test] needs: [unit-test, lint]
runs-on: ubuntu-latest runs-on: ubuntu-latest
steps: steps:
- name: Finish - name: Finish
......
...@@ -8,7 +8,7 @@ docker run --rm \ ...@@ -8,7 +8,7 @@ docker run --rm \
-v "$(pwd)":/sgl-kernel \ -v "$(pwd)":/sgl-kernel \
pytorch/manylinux-builder:cuda${CUDA_VERSION} \ pytorch/manylinux-builder:cuda${CUDA_VERSION} \
bash -c " 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 TORCH_CUDA_ARCH_LIST='7.5 8.0 8.9 9.0+PTX' && \
export CUDA_VERSION=${CUDA_VERSION} && \ export CUDA_VERSION=${CUDA_VERSION} && \
mkdir -p /usr/lib/x86_64-linux-gnu/ && \ mkdir -p /usr/lib/x86_64-linux-gnu/ && \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment