Unverified Commit fb9582c4 authored by alisonshao's avatar alisonshao Committed by GitHub
Browse files

Add multi-GPU configurations to nightly-test.yml (#12585)

parent d22d0447
...@@ -117,7 +117,77 @@ jobs: ...@@ -117,7 +117,77 @@ jobs:
bash scripts/ci/ci_install_dependency.sh bash scripts/ci/ci_install_dependency.sh
- name: Run test - name: Run test
timeout-minutes: 10 timeout-minutes: 60
run: | run: |
cd test/srt cd test/srt
python3 run_suite.py --suite nightly-1-gpu python3 run_suite.py --suite nightly-1-gpu
nightly-test-4-gpu:
if: github.repository == 'sgl-project/sglang'
runs-on: 4-gpu-h100
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Install dependencies
run: |
bash scripts/ci/ci_install_dependency.sh
- name: Run test
timeout-minutes: 30
run: |
cd test/srt
python3 run_suite.py --suite nightly-4-gpu
nightly-test-8-gpu-h200:
if: github.repository == 'sgl-project/sglang'
runs-on: 8-gpu-h200
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Install dependencies
run: |
bash scripts/ci/ci_install_dependency.sh
- name: Run test
timeout-minutes: 30
run: |
cd test/srt
python3 run_suite.py --suite nightly-8-gpu-h200
nightly-test-8-gpu-h20:
if: github.repository == 'sgl-project/sglang'
runs-on: 8-gpu-h20
env:
SGLANG_CI_RDMA_ALL_DEVICES: "mlx5_1,mlx5_2,mlx5_3,mlx5_4"
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Install dependencies
run: |
bash scripts/ci/ci_install_dependency.sh
- name: Run test
timeout-minutes: 30
run: |
cd test/srt
python3 run_suite.py --suite nightly-8-gpu-h20
nightly-test-4-gpu-b200:
if: github.repository == 'sgl-project/sglang'
runs-on: 4-gpu-b200
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Install dependencies
run: |
IS_BLACKWELL=1 bash scripts/ci/ci_install_dependency.sh
- name: Run test
timeout-minutes: 60
run: |
cd test/srt
python3 run_suite.py --suite nightly-4-gpu-b200
...@@ -103,6 +103,11 @@ class SGLangCIAnalyzer: ...@@ -103,6 +103,11 @@ class SGLangCIAnalyzer:
"nightly": [ "nightly": [
"nightly-test-perf-text-models", "nightly-test-perf-text-models",
"nightly-test-eval-text-models", "nightly-test-eval-text-models",
"nightly-test-1-gpu",
"nightly-test-4-gpu",
"nightly-test-8-gpu-h200",
"nightly-test-8-gpu-h20",
"nightly-test-4-gpu-b200",
], ],
"integration": [ "integration": [
"run-all-notebooks", "run-all-notebooks",
...@@ -190,6 +195,10 @@ class SGLangCIAnalyzer: ...@@ -190,6 +195,10 @@ class SGLangCIAnalyzer:
"nightly-test-eval-vlms", "nightly-test-eval-vlms",
"nightly-test-perf-vlms", "nightly-test-perf-vlms",
"nightly-test-1-gpu", "nightly-test-1-gpu",
"nightly-test-4-gpu",
"nightly-test-8-gpu-h200",
"nightly-test-8-gpu-h20",
"nightly-test-4-gpu-b200",
] ]
if job_name in target_jobs: if job_name in target_jobs:
......
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