EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

minor: support flashinfer nightly (#2295)

Browse Source
This commit is contained in:
Yineng Zhang
2024-12-01 18:55:26 +08:00
committed by GitHub
parent 906d795f15
commit fc78640e00
3 changed files with 33 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
.github/workflows/pr-test.yml vendored
View File

@@ -12,6 +12,15 @@ on:
- "python/sglang/**"
- "test/**"
workflow_dispatch:
inputs:
version:
description: "FlashInfer version"
required: true
type: choice
default: 'release'
options:
- 'release'
- 'nightly'
concurrency:
group: pr-test-${{ github.ref }}
@@ -26,6 +35,8 @@ jobs:
uses: actions/checkout@v3
- name: Install dependencies
env:
FLASHINFER_REPO: ${{ inputs.version == 'nightly' && 'https://flashinfer.ai/whl/nightly/cu121/torch2.4/' || 'https://flashinfer.ai/whl/cu121/torch2.4/' }}
run: |
bash scripts/ci_install_dependency.sh
@@ -46,6 +57,8 @@ jobs:
uses: actions/checkout@v3
- name: Install dependencies
env:
FLASHINFER_REPO: ${{ inputs.version == 'nightly' && 'https://flashinfer.ai/whl/nightly/cu121/torch2.4/' || 'https://flashinfer.ai/whl/cu121/torch2.4/' }}
run: |
bash scripts/ci_install_dependency.sh
@@ -66,6 +79,8 @@ jobs:
uses: actions/checkout@v3
- name: Install dependencies
env:
FLASHINFER_REPO: ${{ inputs.version == 'nightly' && 'https://flashinfer.ai/whl/nightly/cu121/torch2.4/' || 'https://flashinfer.ai/whl/cu121/torch2.4/' }}
run: |
bash scripts/ci_install_dependency.sh
@@ -92,6 +107,8 @@ jobs:
uses: actions/checkout@v3
- name: Install dependencies
env:
FLASHINFER_REPO: ${{ inputs.version == 'nightly' && 'https://flashinfer.ai/whl/nightly/cu121/torch2.4/' || 'https://flashinfer.ai/whl/cu121/torch2.4/' }}
run: |
bash scripts/ci_install_dependency.sh
@@ -127,6 +144,8 @@ jobs:
uses: actions/checkout@v3
- name: Install dependencies
env:
FLASHINFER_REPO: ${{ inputs.version == 'nightly' && 'https://flashinfer.ai/whl/nightly/cu121/torch2.4/' || 'https://flashinfer.ai/whl/cu121/torch2.4/' }}
run: |
bash scripts/ci_install_dependency.sh
@@ -156,6 +175,8 @@ jobs:
uses: actions/checkout@v3
- name: Install dependencies
env:
FLASHINFER_REPO: ${{ inputs.version == 'nightly' && 'https://flashinfer.ai/whl/nightly/cu121/torch2.4/' || 'https://flashinfer.ai/whl/cu121/torch2.4/' }}
run: |
bash scripts/ci_install_dependency.sh
@@ -185,6 +206,8 @@ jobs:
uses: actions/checkout@v3
- name: Install dependencies
env:
FLASHINFER_REPO: ${{ inputs.version == 'nightly' && 'https://flashinfer.ai/whl/nightly/cu121/torch2.4/' || 'https://flashinfer.ai/whl/cu121/torch2.4/' }}
run: |
bash scripts/ci_install_dependency.sh
@@ -207,6 +230,8 @@ jobs:
uses: actions/checkout@v3
- name: Install dependencies
env:
FLASHINFER_REPO: ${{ inputs.version == 'nightly' && 'https://flashinfer.ai/whl/nightly/cu121/torch2.4/' || 'https://flashinfer.ai/whl/cu121/torch2.4/' }}
run: |
bash scripts/ci_install_dependency.sh