upgrade torch version for sgl-kernel (#3026)
This commit is contained in:
16
.github/workflows/pr-test-sgl-kernel.yml
vendored
16
.github/workflows/pr-test-sgl-kernel.yml
vendored
@@ -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
|
||||
|
||||
@@ -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/ && \
|
||||
|
||||
Reference in New Issue
Block a user