[NPU]Add a full test pipeline on NPU, resolve issues in the NPU test architecture (#20751)
This commit is contained in:
@@ -0,0 +1,355 @@
|
|||||||
|
name: Full Test (NPU)
|
||||||
|
|
||||||
|
on:
|
||||||
|
# pull_request:
|
||||||
|
# branches:
|
||||||
|
# - main
|
||||||
|
# paths:
|
||||||
|
# - ".github/workflows/full-test-npu.yml"
|
||||||
|
workflow_dispatch:
|
||||||
|
inputs:
|
||||||
|
ref:
|
||||||
|
description: 'Git ref (branch, tag, or SHA) to test. If not provided, uses the default branch.'
|
||||||
|
required: false
|
||||||
|
type: string
|
||||||
|
default: ''
|
||||||
|
job_filter:
|
||||||
|
description: 'Select which job to run (leave empty or "all" to run all jobs)'
|
||||||
|
required: false
|
||||||
|
type: string
|
||||||
|
default: 'all'
|
||||||
|
image_a3:
|
||||||
|
description: 'The a3 running docker image of the test task.'
|
||||||
|
required: false
|
||||||
|
type: string
|
||||||
|
default: 'swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11'
|
||||||
|
skip_install_flag:
|
||||||
|
description: 'Indicates whether to skip the installation of sglang, defaulting to false.'
|
||||||
|
required: false
|
||||||
|
type: string
|
||||||
|
default: 'false'
|
||||||
|
|
||||||
|
concurrency:
|
||||||
|
group: full-test-npu-${{ inputs.ref || github.ref }}
|
||||||
|
cancel-in-progress: ${{ github.event_name != 'workflow_call' }}
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
set-image-config:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
outputs:
|
||||||
|
ref: ${{ steps.set-vars.outputs.ref }}
|
||||||
|
job_filter: ${{ steps.set-vars.outputs.job_filter }}
|
||||||
|
image_a3: ${{ steps.set-vars.outputs.image_a3 }}
|
||||||
|
skip_install_flag: ${{ steps.set-vars.outputs.skip_install_flag }}
|
||||||
|
steps:
|
||||||
|
# When triggered by PR, no inputs parameters are used. The latest community code is tested by default.
|
||||||
|
- name: Set image config
|
||||||
|
id: set-vars
|
||||||
|
run: |
|
||||||
|
if [ -z "${{ inputs.ref }}" ]; then
|
||||||
|
echo "ref=" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "ref=${{ inputs.ref }}" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "${{ inputs.job_filter }}" ]; then
|
||||||
|
echo "job_filter=all" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "job_filter=${{ inputs.job_filter }}" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "${{ inputs.image_a3 }}" ]; then
|
||||||
|
echo "image_a3=swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "image_a3=${{ inputs.image_a3 }}" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "${{ inputs.skip_install_flag }}" ]; then
|
||||||
|
echo "skip_install_flag=false" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "skip_install_flag=${{ inputs.skip_install_flag }}" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
|
nighly-test-npu:
|
||||||
|
needs: [set-image-config]
|
||||||
|
name: nightly-test-npu
|
||||||
|
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
||||||
|
uses: ./.github/workflows/nightly-test-npu.yml
|
||||||
|
with:
|
||||||
|
ref: ${{ needs.set-image-config.outputs.ref }}
|
||||||
|
job_filter: ${{ needs.set-image-config.outputs.job_filter }}
|
||||||
|
image_a3: ${{ needs.set-image-config.outputs.image_a3 }}
|
||||||
|
skip_install_flag: ${{ needs.set-image-config.outputs.skip_install_flag }}
|
||||||
|
secrets: inherit
|
||||||
|
|
||||||
|
full-1-npu-a3:
|
||||||
|
needs: [set-image-config]
|
||||||
|
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
||||||
|
runs-on: linux-aarch64-a3-2
|
||||||
|
container:
|
||||||
|
image: ${{ needs.set-image-config.outputs.image_a3 }}
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
ref: ${{ needs.set-image-config.outputs.ref || github.ref }}
|
||||||
|
|
||||||
|
- name: Install dependencies
|
||||||
|
env:
|
||||||
|
TORCH_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/whl/cpu"
|
||||||
|
PYPI_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple"
|
||||||
|
GITHUB_PROXY_URL: "https://gh-proxy.test.osinfra.cn/"
|
||||||
|
run: |
|
||||||
|
# speed up by using infra cache services
|
||||||
|
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
||||||
|
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
||||||
|
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
||||||
|
pip config set global.trusted-host "${CACHING_URL}"
|
||||||
|
|
||||||
|
if [ ${{ needs.set-image-config.outputs.skip_install_flag }} != "true" ];then
|
||||||
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
|
fi
|
||||||
|
|
||||||
|
# copy required file from our daily cache
|
||||||
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
|
# copy gsm8k dataset
|
||||||
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
|
- name: Print Log Information
|
||||||
|
run: |
|
||||||
|
bash scripts/ci/npu/npu_log_print.sh
|
||||||
|
|
||||||
|
- name: Run test
|
||||||
|
timeout-minutes: 240
|
||||||
|
env:
|
||||||
|
SGLANG_USE_MODELSCOPE: true
|
||||||
|
SGLANG_IS_IN_CI: true
|
||||||
|
HF_ENDPOINT: https://hf-mirror.com
|
||||||
|
TORCH_EXTENSIONS_DIR: /tmp/torch_extensions
|
||||||
|
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
||||||
|
STREAMS_PER_DEVICE: 32
|
||||||
|
run: |
|
||||||
|
pip install sglang_router
|
||||||
|
hf download lmms-lab/MMMU --repo-type dataset
|
||||||
|
pip install sentence_transformers torchaudio==2.8.0
|
||||||
|
pip install protobuf==6.31.1 zss pre-commit wandb>=0.16.0 tenacity==8.3.0 loguru openpyxl latex2sympy2 zstandard transformers-stream-generator tqdm-multiprocess pycocoevalcap
|
||||||
|
pip install yt-dlp sentencepiece==0.1.99 nltk av ftfy sqlitedict==2.1.0 sacrebleu>=1.5.0 pytablewriter black==24.1.0 isort==5.13.2 peft>=0.2.0 accelerate>=0.29.1
|
||||||
|
pip install jsonlines httpx==0.25.0 evaluate>=0.4.0 datasets==2.16.1 numexpr xgrammar==0.1.25 numpy==1.26.4 dotenv
|
||||||
|
git clone --branch v0.3.3 --depth 1 https://github.com/EvolvingLMMs-Lab/lmms-eval.git
|
||||||
|
cd ./lmms-eval
|
||||||
|
nohup pip install . > lmmslog.txt 2>&1 &
|
||||||
|
sleep 120
|
||||||
|
export PYTHONPATH=$PYTHONPATH:$(pwd)
|
||||||
|
cd ../
|
||||||
|
cd test
|
||||||
|
python3 run_suite.py --hw npu --suite full-1-npu-a3 --nightly --continue-on-error --timeout-per-file 3600
|
||||||
|
|
||||||
|
full-2-npu-a3:
|
||||||
|
needs: [set-image-config]
|
||||||
|
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
||||||
|
runs-on: linux-aarch64-a3-2
|
||||||
|
container:
|
||||||
|
image: ${{ needs.set-image-config.outputs.image_a3 }}
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
ref: ${{ needs.set-image-config.outputs.ref || github.ref }}
|
||||||
|
|
||||||
|
- name: Install dependencies
|
||||||
|
env:
|
||||||
|
TORCH_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/whl/cpu"
|
||||||
|
PYPI_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple"
|
||||||
|
GITHUB_PROXY_URL: "https://gh-proxy.test.osinfra.cn/"
|
||||||
|
run: |
|
||||||
|
# speed up by using infra cache services
|
||||||
|
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
||||||
|
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
||||||
|
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
||||||
|
pip config set global.trusted-host "${CACHING_URL}"
|
||||||
|
|
||||||
|
if [ ${{ needs.set-image-config.outputs.skip_install_flag }} != "true" ];then
|
||||||
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
|
fi
|
||||||
|
|
||||||
|
# copy required file from our daily cache
|
||||||
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
|
# copy gsm8k dataset
|
||||||
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
|
- name: Print Log Information
|
||||||
|
run: |
|
||||||
|
bash scripts/ci/npu/npu_log_print.sh
|
||||||
|
|
||||||
|
- name: Run test
|
||||||
|
timeout-minutes: 240
|
||||||
|
env:
|
||||||
|
SGLANG_USE_MODELSCOPE: true
|
||||||
|
SGLANG_IS_IN_CI: true
|
||||||
|
HF_ENDPOINT: https://hf-mirror.com
|
||||||
|
TORCH_EXTENSIONS_DIR: /tmp/torch_extensions
|
||||||
|
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
||||||
|
STREAMS_PER_DEVICE: 32
|
||||||
|
run: |
|
||||||
|
pip install sglang_router
|
||||||
|
hf download lmms-lab/MMMU --repo-type dataset
|
||||||
|
pip install sentence_transformers torchaudio==2.8.0
|
||||||
|
pip install protobuf==6.31.1 zss pre-commit wandb>=0.16.0 tenacity==8.3.0 loguru openpyxl latex2sympy2 zstandard transformers-stream-generator tqdm-multiprocess pycocoevalcap
|
||||||
|
pip install yt-dlp sentencepiece==0.1.99 nltk av ftfy sqlitedict==2.1.0 sacrebleu>=1.5.0 pytablewriter black==24.1.0 isort==5.13.2 peft>=0.2.0 accelerate>=0.29.1
|
||||||
|
pip install jsonlines httpx==0.25.0 evaluate>=0.4.0 datasets==2.16.1 numexpr xgrammar==0.1.25 numpy==1.26.4 dotenv
|
||||||
|
git clone --branch v0.3.3 --depth 1 https://github.com/EvolvingLMMs-Lab/lmms-eval.git
|
||||||
|
cd ./lmms-eval
|
||||||
|
nohup pip install . > lmmslog.txt 2>&1 &
|
||||||
|
sleep 120
|
||||||
|
export PYTHONPATH=$PYTHONPATH:$(pwd)
|
||||||
|
cd ../
|
||||||
|
cd test
|
||||||
|
python3 run_suite.py --hw npu --suite full-2-npu-a3 --nightly --continue-on-error --timeout-per-file 3600
|
||||||
|
|
||||||
|
full-4-npu-a3:
|
||||||
|
needs: [set-image-config]
|
||||||
|
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
||||||
|
runs-on: linux-aarch64-a3-4
|
||||||
|
container:
|
||||||
|
image: ${{ needs.set-image-config.outputs.image_a3 }}
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
ref: ${{ needs.set-image-config.outputs.ref || github.ref }}
|
||||||
|
|
||||||
|
- name: Install dependencies
|
||||||
|
env:
|
||||||
|
TORCH_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/whl/cpu"
|
||||||
|
PYPI_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple"
|
||||||
|
GITHUB_PROXY_URL: "https://gh-proxy.test.osinfra.cn/"
|
||||||
|
run: |
|
||||||
|
# speed up by using infra cache services
|
||||||
|
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
||||||
|
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
||||||
|
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
||||||
|
pip config set global.trusted-host "${CACHING_URL}"
|
||||||
|
|
||||||
|
if [ ${{ needs.set-image-config.outputs.skip_install_flag }} != "true" ];then
|
||||||
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
|
fi
|
||||||
|
|
||||||
|
# copy required file from our daily cache
|
||||||
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
|
# copy gsm8k dataset
|
||||||
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
|
- name: Print Log Information
|
||||||
|
run: |
|
||||||
|
bash scripts/ci/npu/npu_log_print.sh
|
||||||
|
|
||||||
|
- name: Run test
|
||||||
|
timeout-minutes: 240
|
||||||
|
env:
|
||||||
|
SGLANG_USE_MODELSCOPE: true
|
||||||
|
SGLANG_IS_IN_CI: true
|
||||||
|
HF_ENDPOINT: https://hf-mirror.com
|
||||||
|
TORCH_EXTENSIONS_DIR: /tmp/torch_extensions
|
||||||
|
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
||||||
|
STREAMS_PER_DEVICE: 32
|
||||||
|
run: |
|
||||||
|
pip install sglang_router
|
||||||
|
hf download lmms-lab/MMMU --repo-type dataset
|
||||||
|
pip install sentence_transformers torchaudio==2.8.0
|
||||||
|
pip install protobuf==6.31.1 zss pre-commit wandb>=0.16.0 tenacity==8.3.0 loguru openpyxl latex2sympy2 zstandard transformers-stream-generator tqdm-multiprocess pycocoevalcap
|
||||||
|
pip install yt-dlp sentencepiece==0.1.99 nltk av ftfy sqlitedict==2.1.0 sacrebleu>=1.5.0 pytablewriter black==24.1.0 isort==5.13.2 peft>=0.2.0 accelerate>=0.29.1
|
||||||
|
pip install jsonlines httpx==0.25.0 evaluate>=0.4.0 datasets==2.16.1 numexpr xgrammar==0.1.25 numpy==1.26.4 dotenv
|
||||||
|
git clone --branch v0.3.3 --depth 1 https://github.com/EvolvingLMMs-Lab/lmms-eval.git
|
||||||
|
cd ./lmms-eval
|
||||||
|
nohup pip install . > lmmslog.txt 2>&1 &
|
||||||
|
sleep 120
|
||||||
|
export PYTHONPATH=$PYTHONPATH:$(pwd)
|
||||||
|
cd ../
|
||||||
|
cd test
|
||||||
|
python3 run_suite.py --hw npu --suite full-4-npu-a3 --nightly --continue-on-error --timeout-per-file 3600
|
||||||
|
|
||||||
|
full-16-npu-a3:
|
||||||
|
needs: [set-image-config]
|
||||||
|
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
||||||
|
runs-on: linux-aarch64-a3-16
|
||||||
|
container:
|
||||||
|
image: ${{ needs.set-image-config.outputs.image_a3 }}
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
ref: ${{ needs.set-image-config.outputs.ref || github.ref }}
|
||||||
|
|
||||||
|
- name: Install dependencies
|
||||||
|
env:
|
||||||
|
TORCH_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/whl/cpu"
|
||||||
|
PYPI_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple"
|
||||||
|
GITHUB_PROXY_URL: "https://gh-proxy.test.osinfra.cn/"
|
||||||
|
run: |
|
||||||
|
# speed up by using infra cache services
|
||||||
|
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
||||||
|
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
||||||
|
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
||||||
|
pip config set global.trusted-host "${CACHING_URL}"
|
||||||
|
|
||||||
|
if [ ${{ needs.set-image-config.outputs.skip_install_flag }} != "true" ];then
|
||||||
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
|
fi
|
||||||
|
|
||||||
|
# copy required file from our daily cache
|
||||||
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
|
# copy gsm8k dataset
|
||||||
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
|
- name: Print Log Information
|
||||||
|
run: |
|
||||||
|
bash scripts/ci/npu/npu_log_print.sh
|
||||||
|
|
||||||
|
- name: Run test
|
||||||
|
timeout-minutes: 240
|
||||||
|
env:
|
||||||
|
SGLANG_USE_MODELSCOPE: true
|
||||||
|
SGLANG_IS_IN_CI: true
|
||||||
|
HF_ENDPOINT: https://hf-mirror.com
|
||||||
|
TORCH_EXTENSIONS_DIR: /tmp/torch_extensions
|
||||||
|
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
||||||
|
STREAMS_PER_DEVICE: 32
|
||||||
|
run: |
|
||||||
|
pip install sglang_router
|
||||||
|
hf download lmms-lab/MMMU --repo-type dataset
|
||||||
|
pip install sentence_transformers torchaudio==2.8.0
|
||||||
|
pip install protobuf==6.31.1 zss pre-commit wandb>=0.16.0 tenacity==8.3.0 loguru openpyxl latex2sympy2 zstandard transformers-stream-generator tqdm-multiprocess pycocoevalcap
|
||||||
|
pip install yt-dlp sentencepiece==0.1.99 nltk av ftfy sqlitedict==2.1.0 sacrebleu>=1.5.0 pytablewriter black==24.1.0 isort==5.13.2 peft>=0.2.0 accelerate>=0.29.1
|
||||||
|
pip install jsonlines httpx==0.25.0 evaluate>=0.4.0 datasets==2.16.1 numexpr xgrammar==0.1.25 numpy==1.26.4 dotenv
|
||||||
|
git clone --branch v0.3.3 --depth 1 https://github.com/EvolvingLMMs-Lab/lmms-eval.git
|
||||||
|
cd ./lmms-eval
|
||||||
|
nohup pip install . > lmmslog.txt 2>&1 &
|
||||||
|
sleep 120
|
||||||
|
export PYTHONPATH=$PYTHONPATH:$(pwd)
|
||||||
|
cd ../
|
||||||
|
cd test
|
||||||
|
python3 run_suite.py --hw npu --suite full-16-npu-a3 --nightly --continue-on-error --timeout-per-file 3600
|
||||||
|
|
||||||
|
check-all-jobs:
|
||||||
|
if: github.repository == 'sgl-project/sglang' && always()
|
||||||
|
needs:
|
||||||
|
- nighly-test-npu
|
||||||
|
- full-1-npu-a3
|
||||||
|
- full-2-npu-a3
|
||||||
|
- full-4-npu-a3
|
||||||
|
- full-16-npu-a3
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
container:
|
||||||
|
image: docker.m.daocloud.io/ubuntu:22.04
|
||||||
|
steps:
|
||||||
|
- name: Check if any job failed
|
||||||
|
run: |
|
||||||
|
if [[ "${{ contains(needs.*.result, 'failure') }}" == "true" ]]; then
|
||||||
|
echo "One or more nightly test jobs failed"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
if [[ "${{ contains(needs.*.result, 'cancelled') }}" == "true" ]]; then
|
||||||
|
echo "One or more nightly test jobs were cancelled"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
echo "All nightly test jobs passed"
|
||||||
@@ -2,7 +2,7 @@ name: Nightly Test (NPU)
|
|||||||
|
|
||||||
on:
|
on:
|
||||||
schedule:
|
schedule:
|
||||||
- cron: '0 17 * * *' # Execute at 1:00 a.m. Beijing Time every day
|
- cron: '0 18 * * *' # Execute at 2:00 a.m. Beijing Time every day
|
||||||
pull_request:
|
pull_request:
|
||||||
branches:
|
branches:
|
||||||
- main
|
- main
|
||||||
@@ -21,13 +21,61 @@ on:
|
|||||||
required: false
|
required: false
|
||||||
type: string
|
type: string
|
||||||
default: 'all'
|
default: 'all'
|
||||||
|
image_a3:
|
||||||
|
description: 'The a3 running docker image of the test task.'
|
||||||
|
required: false
|
||||||
|
type: string
|
||||||
|
default: 'swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11'
|
||||||
|
skip_install_flag:
|
||||||
|
description: 'Indicates whether to skip the installation of sglang, defaulting to false.'
|
||||||
|
required: false
|
||||||
|
type: string
|
||||||
|
default: 'false'
|
||||||
|
|
||||||
|
|
||||||
concurrency:
|
concurrency:
|
||||||
group: nightly-test-npu-${{ inputs.ref || github.ref }}
|
group: nightly-test-npu-${{ inputs.ref || github.ref }}
|
||||||
cancel-in-progress: ${{ github.event_name != 'workflow_call' }}
|
cancel-in-progress: ${{ github.event_name != 'workflow_call' }}
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
|
set-image-config:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
outputs:
|
||||||
|
ref: ${{ steps.set-vars.outputs.ref }}
|
||||||
|
job_filter: ${{ steps.set-vars.outputs.job_filter }}
|
||||||
|
image_a3: ${{ steps.set-vars.outputs.image_a3 }}
|
||||||
|
skip_install_flag: ${{ steps.set-vars.outputs.skip_install_flag }}
|
||||||
|
steps:
|
||||||
|
# When triggered by PR, no inputs parameters are used. The latest community code is tested by default.
|
||||||
|
- name: Set image config
|
||||||
|
id: set-vars
|
||||||
|
run: |
|
||||||
|
if [ -z "${{ inputs.ref }}" ]; then
|
||||||
|
echo "ref=" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "ref=${{ inputs.ref }}" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "${{ inputs.job_filter }}" ]; then
|
||||||
|
echo "job_filter=all" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "job_filter=${{ inputs.job_filter }}" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "${{ inputs.image_a3 }}" ]; then
|
||||||
|
echo "image_a3=swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "image_a3=${{ inputs.image_a3 }}" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "${{ inputs.skip_install_flag }}" ]; then
|
||||||
|
echo "skip_install_flag=false" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "skip_install_flag=${{ inputs.skip_install_flag }}" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
nightly-1-npu-a3:
|
nightly-1-npu-a3:
|
||||||
|
needs: [set-image-config]
|
||||||
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
||||||
runs-on: linux-aarch64-a3-2
|
runs-on: linux-aarch64-a3-2
|
||||||
strategy:
|
strategy:
|
||||||
@@ -35,26 +83,33 @@ jobs:
|
|||||||
matrix:
|
matrix:
|
||||||
part: [0, 1]
|
part: [0, 1]
|
||||||
container:
|
container:
|
||||||
image: swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11
|
image: ${{ needs.set-image-config.outputs.image_a3 }}
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout code
|
- name: Checkout code
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
ref: ${{ inputs.ref || github.ref }}
|
ref: ${{ needs.set-image-config.outputs.ref || github.ref }}
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
|
env:
|
||||||
|
TORCH_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/whl/cpu"
|
||||||
|
PYPI_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple"
|
||||||
|
GITHUB_PROXY_URL: "https://gh-proxy.test.osinfra.cn/"
|
||||||
run: |
|
run: |
|
||||||
# speed up by using infra cache services
|
# speed up by using infra cache services
|
||||||
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
||||||
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
||||||
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
||||||
pip config set global.extra-index-url "https://pypi.tuna.tsinghua.edu.cn/simple"
|
pip config set global.trusted-host "${CACHING_URL}"
|
||||||
pip config set global.trusted-host "${CACHING_URL} pypi.tuna.tsinghua.edu.cn"
|
|
||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
if [ ${{ needs.set-image-config.outputs.skip_install_flag }} != "true" ];then
|
||||||
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
|
fi
|
||||||
|
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Print Log Information
|
- name: Print Log Information
|
||||||
run: |
|
run: |
|
||||||
@@ -86,6 +141,7 @@ jobs:
|
|||||||
python3 run_suite.py --hw npu --suite nightly-1-npu-a3 --nightly --continue-on-error --timeout-per-file 3600 --auto-partition-id ${{ matrix.part }} --auto-partition-size 2
|
python3 run_suite.py --hw npu --suite nightly-1-npu-a3 --nightly --continue-on-error --timeout-per-file 3600 --auto-partition-id ${{ matrix.part }} --auto-partition-size 2
|
||||||
|
|
||||||
nightly-2-npu-a3:
|
nightly-2-npu-a3:
|
||||||
|
needs: [set-image-config]
|
||||||
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
||||||
runs-on: linux-aarch64-a3-2
|
runs-on: linux-aarch64-a3-2
|
||||||
strategy:
|
strategy:
|
||||||
@@ -93,26 +149,33 @@ jobs:
|
|||||||
matrix:
|
matrix:
|
||||||
part: [0]
|
part: [0]
|
||||||
container:
|
container:
|
||||||
image: swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11
|
image: ${{ needs.set-image-config.outputs.image_a3 }}
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout code
|
- name: Checkout code
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
ref: ${{ inputs.ref || github.ref }}
|
ref: ${{ needs.set-image-config.outputs.ref || github.ref }}
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
|
env:
|
||||||
|
TORCH_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/whl/cpu"
|
||||||
|
PYPI_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple"
|
||||||
|
GITHUB_PROXY_URL: "https://gh-proxy.test.osinfra.cn/"
|
||||||
run: |
|
run: |
|
||||||
# speed up by using infra cache services
|
# speed up by using infra cache services
|
||||||
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
||||||
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
||||||
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
||||||
pip config set global.extra-index-url "https://pypi.tuna.tsinghua.edu.cn/simple"
|
pip config set global.trusted-host "${CACHING_URL}"
|
||||||
pip config set global.trusted-host "${CACHING_URL} pypi.tuna.tsinghua.edu.cn"
|
|
||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
if [ ${{ needs.set-image-config.outputs.skip_install_flag }} != "true" ];then
|
||||||
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
|
fi
|
||||||
|
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Print Log Information
|
- name: Print Log Information
|
||||||
run: |
|
run: |
|
||||||
@@ -143,6 +206,7 @@ jobs:
|
|||||||
python3 run_suite.py --hw npu --suite nightly-2-npu-a3 --nightly --continue-on-error --timeout-per-file 3600 --auto-partition-id ${{ matrix.part }} --auto-partition-size 1
|
python3 run_suite.py --hw npu --suite nightly-2-npu-a3 --nightly --continue-on-error --timeout-per-file 3600 --auto-partition-id ${{ matrix.part }} --auto-partition-size 1
|
||||||
|
|
||||||
nightly-4-npu-a3:
|
nightly-4-npu-a3:
|
||||||
|
needs: [set-image-config]
|
||||||
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
||||||
runs-on: linux-aarch64-a3-4
|
runs-on: linux-aarch64-a3-4
|
||||||
strategy:
|
strategy:
|
||||||
@@ -150,25 +214,33 @@ jobs:
|
|||||||
matrix:
|
matrix:
|
||||||
part: [0]
|
part: [0]
|
||||||
container:
|
container:
|
||||||
image: swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11
|
image: ${{ needs.set-image-config.outputs.image_a3 }}
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout code
|
- name: Checkout code
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
ref: ${{ inputs.ref || github.ref }}
|
ref: ${{ needs.set-image-config.outputs.ref|| github.ref }}
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
|
env:
|
||||||
|
TORCH_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/whl/cpu"
|
||||||
|
PYPI_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple"
|
||||||
|
GITHUB_PROXY_URL: "https://gh-proxy.test.osinfra.cn/"
|
||||||
run: |
|
run: |
|
||||||
# speed up by using infra cache services
|
# speed up by using infra cache services
|
||||||
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
||||||
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
||||||
pip config set global.extra-index-url "https://pypi.tuna.tsinghua.edu.cn/simple"
|
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
||||||
pip config set global.trusted-host "${CACHING_URL} pypi.tuna.tsinghua.edu.cn"
|
pip config set global.trusted-host "${CACHING_URL}"
|
||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
|
||||||
|
if [ ${{ needs.set-image-config.outputs.skip_install_flag }} != "true" ];then
|
||||||
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
|
fi
|
||||||
|
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Print Log Information
|
- name: Print Log Information
|
||||||
run: |
|
run: |
|
||||||
@@ -200,6 +272,7 @@ jobs:
|
|||||||
python3 run_suite.py --hw npu --suite nightly-4-npu-a3 --nightly --continue-on-error --timeout-per-file 3600 --auto-partition-id ${{ matrix.part }} --auto-partition-size 1
|
python3 run_suite.py --hw npu --suite nightly-4-npu-a3 --nightly --continue-on-error --timeout-per-file 3600 --auto-partition-id ${{ matrix.part }} --auto-partition-size 1
|
||||||
|
|
||||||
nightly-8-npu-a3:
|
nightly-8-npu-a3:
|
||||||
|
needs: [set-image-config]
|
||||||
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
||||||
runs-on: linux-aarch64-a3-8
|
runs-on: linux-aarch64-a3-8
|
||||||
strategy:
|
strategy:
|
||||||
@@ -207,26 +280,33 @@ jobs:
|
|||||||
matrix:
|
matrix:
|
||||||
part: [0]
|
part: [0]
|
||||||
container:
|
container:
|
||||||
image: swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11
|
image: ${{ needs.set-image-config.outputs.image_a3 }}
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout code
|
- name: Checkout code
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
ref: ${{ inputs.ref || github.ref }}
|
ref: ${{ needs.set-image-config.outputs.ref || github.ref }}
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
|
env:
|
||||||
|
TORCH_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/whl/cpu"
|
||||||
|
PYPI_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple"
|
||||||
|
GITHUB_PROXY_URL: "https://gh-proxy.test.osinfra.cn/"
|
||||||
run: |
|
run: |
|
||||||
# speed up by using infra cache services
|
# speed up by using infra cache services
|
||||||
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
||||||
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
||||||
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
||||||
pip config set global.extra-index-url "https://pypi.tuna.tsinghua.edu.cn/simple"
|
pip config set global.trusted-host "${CACHING_URL}"
|
||||||
pip config set global.trusted-host "${CACHING_URL} pypi.tuna.tsinghua.edu.cn"
|
|
||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
if [ ${{ needs.set-image-config.outputs.skip_install_flag }} != "true" ];then
|
||||||
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
|
fi
|
||||||
|
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://gh-proxy.test.osinfra.cn/https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Print Log Information
|
- name: Print Log Information
|
||||||
run: |
|
run: |
|
||||||
@@ -258,6 +338,7 @@ jobs:
|
|||||||
python3 run_suite.py --hw npu --suite nightly-8-npu-a3 --nightly --continue-on-error --timeout-per-file 3600 --auto-partition-id ${{ matrix.part }} --auto-partition-size 1
|
python3 run_suite.py --hw npu --suite nightly-8-npu-a3 --nightly --continue-on-error --timeout-per-file 3600 --auto-partition-id ${{ matrix.part }} --auto-partition-size 1
|
||||||
|
|
||||||
nightly-16-npu-a3:
|
nightly-16-npu-a3:
|
||||||
|
needs: [set-image-config]
|
||||||
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
if: ${{ (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') }}
|
||||||
runs-on: linux-aarch64-a3-16
|
runs-on: linux-aarch64-a3-16
|
||||||
strategy:
|
strategy:
|
||||||
@@ -265,26 +346,33 @@ jobs:
|
|||||||
matrix:
|
matrix:
|
||||||
part: [0, 1]
|
part: [0, 1]
|
||||||
container:
|
container:
|
||||||
image: swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11
|
image: ${{ needs.set-image-config.outputs.image_a3 }}
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout code
|
- name: Checkout code
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
ref: ${{ inputs.ref || github.ref }}
|
ref: ${{ needs.set-image-config.outputs.ref || github.ref }}
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
|
env:
|
||||||
|
TORCH_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/whl/cpu"
|
||||||
|
PYPI_CACHE_URL: "http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple"
|
||||||
|
GITHUB_PROXY_URL: "https://gh-proxy.test.osinfra.cn/"
|
||||||
run: |
|
run: |
|
||||||
# speed up by using infra cache services
|
# speed up by using infra cache services
|
||||||
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
CACHING_URL="cache-service.nginx-pypi-cache.svc.cluster.local"
|
||||||
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
sed -Ei "s@(ports|archive).ubuntu.com@${CACHING_URL}:8081@g" /etc/apt/sources.list
|
||||||
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
pip config set global.index-url http://${CACHING_URL}/pypi/simple
|
||||||
pip config set global.extra-index-url "https://pypi.tuna.tsinghua.edu.cn/simple"
|
pip config set global.trusted-host "${CACHING_URL}"
|
||||||
pip config set global.trusted-host "${CACHING_URL} pypi.tuna.tsinghua.edu.cn"
|
|
||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
if [ ${{ needs.set-image-config.outputs.skip_install_flag }} != "true" ];then
|
||||||
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
|
fi
|
||||||
|
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://gh-proxy.test.osinfra.cn/https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Print Log Information
|
- name: Print Log Information
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
@@ -76,7 +76,7 @@ jobs:
|
|||||||
uses: ./.github/workflows/pr-gate.yml
|
uses: ./.github/workflows/pr-gate.yml
|
||||||
secrets: inherit
|
secrets: inherit
|
||||||
|
|
||||||
per-commit-1-npu-a2:
|
stage-b-test-1-npu-a2:
|
||||||
needs: [check-changes, pr-gate]
|
needs: [check-changes, pr-gate]
|
||||||
if: needs.check-changes.outputs.main_package == 'true'
|
if: needs.check-changes.outputs.main_package == 'true'
|
||||||
runs-on: linux-aarch64-a2-1
|
runs-on: linux-aarch64-a2-1
|
||||||
@@ -111,21 +111,8 @@ jobs:
|
|||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh 910b
|
bash scripts/ci/npu/npu_ci_install_dependency.sh 910b
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Run registered test
|
|
||||||
timeout-minutes: 240
|
|
||||||
env:
|
|
||||||
SGLANG_USE_MODELSCOPE: true
|
|
||||||
SGLANG_IS_IN_CI: true
|
|
||||||
HF_ENDPOINT: https://hf-mirror.com
|
|
||||||
TORCH_EXTENSIONS_DIR: /tmp/torch_extensions
|
|
||||||
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
|
||||||
STREAMS_PER_DEVICE: 32
|
|
||||||
run: |
|
|
||||||
cd test
|
|
||||||
python3 run_suite.py --hw npu --suite per-commit-1-npu-a2 --continue-on-error --timeout-per-file 3600 --auto-partition-id ${{ matrix.part }} --auto-partition-size 2
|
|
||||||
|
|
||||||
- name: Run test
|
- name: Run test
|
||||||
timeout-minutes: 60
|
timeout-minutes: 60
|
||||||
@@ -137,10 +124,10 @@ jobs:
|
|||||||
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
||||||
STREAMS_PER_DEVICE: 32
|
STREAMS_PER_DEVICE: 32
|
||||||
run: |
|
run: |
|
||||||
cd test/srt
|
cd test
|
||||||
python3 run_suite.py --suite per-commit-1-npu-a2 --auto-partition-id ${{ matrix.part }} --auto-partition-size 2
|
python3 run_suite.py --hw npu --suite stage-b-test-1-npu-a2 --auto-partition-id ${{ matrix.part }} --auto-partition-size 2
|
||||||
|
|
||||||
per-commit-2-npu-a2:
|
stage-b-test-2-npu-a2:
|
||||||
needs: [check-changes, pr-gate]
|
needs: [check-changes, pr-gate]
|
||||||
if: needs.check-changes.outputs.main_package == 'true'
|
if: needs.check-changes.outputs.main_package == 'true'
|
||||||
runs-on: linux-aarch64-a2-2
|
runs-on: linux-aarch64-a2-2
|
||||||
@@ -175,8 +162,8 @@ jobs:
|
|||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh 910b
|
bash scripts/ci/npu/npu_ci_install_dependency.sh 910b
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Run test
|
- name: Run test
|
||||||
timeout-minutes: 60
|
timeout-minutes: 60
|
||||||
@@ -188,10 +175,10 @@ jobs:
|
|||||||
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
||||||
STREAMS_PER_DEVICE: 32
|
STREAMS_PER_DEVICE: 32
|
||||||
run: |
|
run: |
|
||||||
cd test/srt
|
cd test
|
||||||
python3 run_suite.py --suite per-commit-2-npu-a2 --auto-partition-id ${{ matrix.part }} --auto-partition-size 2
|
python3 run_suite.py --hw npu --suite stage-b-test-2-npu-a2 --auto-partition-id ${{ matrix.part }} --auto-partition-size 2
|
||||||
|
|
||||||
per-commit-4-npu-a3:
|
stage-b-test-4-npu-a3:
|
||||||
needs: [check-changes, pr-gate]
|
needs: [check-changes, pr-gate]
|
||||||
if: needs.check-changes.outputs.main_package == 'true'
|
if: needs.check-changes.outputs.main_package == 'true'
|
||||||
runs-on: linux-aarch64-a3-4
|
runs-on: linux-aarch64-a3-4
|
||||||
@@ -222,8 +209,8 @@ jobs:
|
|||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Run test
|
- name: Run test
|
||||||
timeout-minutes: 60
|
timeout-minutes: 60
|
||||||
@@ -235,10 +222,11 @@ jobs:
|
|||||||
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
||||||
STREAMS_PER_DEVICE: 32
|
STREAMS_PER_DEVICE: 32
|
||||||
run: |
|
run: |
|
||||||
cd test/srt
|
cd test
|
||||||
python3 run_suite.py --suite per-commit-4-npu-a3 --timeout-per-file 3600
|
python3 run_suite.py --hw npu --suite stage-b-test-4-npu-a3 --timeout-per-file 3600
|
||||||
|
|
||||||
per-commit-16-npu-a3:
|
|
||||||
|
stage-b-test-16-npu-a3:
|
||||||
needs: [check-changes, pr-gate]
|
needs: [check-changes, pr-gate]
|
||||||
if: needs.check-changes.outputs.main_package == 'true'
|
if: needs.check-changes.outputs.main_package == 'true'
|
||||||
runs-on: linux-aarch64-a3-16
|
runs-on: linux-aarch64-a3-16
|
||||||
@@ -269,8 +257,8 @@ jobs:
|
|||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://gh-proxy.test.osinfra.cn/https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Run test
|
- name: Run test
|
||||||
timeout-minutes: 60
|
timeout-minutes: 60
|
||||||
@@ -282,8 +270,8 @@ jobs:
|
|||||||
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
PYTORCH_NPU_ALLOC_CONF: "expandable_segments:True"
|
||||||
STREAMS_PER_DEVICE: 32
|
STREAMS_PER_DEVICE: 32
|
||||||
run: |
|
run: |
|
||||||
cd test/srt
|
cd test
|
||||||
python3 run_suite.py --suite per-commit-16-npu-a3 --timeout-per-file 3600
|
python3 run_suite.py --hw npu --suite stage-b-test-16-npu-a3 --timeout-per-file 3600
|
||||||
|
|
||||||
multimodal-gen-test-1-npu-a3:
|
multimodal-gen-test-1-npu-a3:
|
||||||
needs: [check-changes, pr-gate]
|
needs: [check-changes, pr-gate]
|
||||||
@@ -314,8 +302,8 @@ jobs:
|
|||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh a3 diffusion
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3 diffusion
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://gh-proxy.test.osinfra.cn/https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Run test
|
- name: Run test
|
||||||
timeout-minutes: 60
|
timeout-minutes: 60
|
||||||
@@ -360,8 +348,8 @@ jobs:
|
|||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh a3 diffusion
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3 diffusion
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://gh-proxy.test.osinfra.cn/https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Run test
|
- name: Run test
|
||||||
timeout-minutes: 60
|
timeout-minutes: 60
|
||||||
@@ -380,7 +368,7 @@ jobs:
|
|||||||
multimodal-gen-test-8-npu-a3:
|
multimodal-gen-test-8-npu-a3:
|
||||||
needs: [check-changes, pr-gate]
|
needs: [check-changes, pr-gate]
|
||||||
if: needs.check-changes.outputs.multimodal_gen == 'true'
|
if: needs.check-changes.outputs.multimodal_gen == 'true'
|
||||||
runs-on: linux-aarch64-a3-16
|
runs-on: linux-aarch64-a3-8
|
||||||
container:
|
container:
|
||||||
image: swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11
|
image: swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.5.0-a3-ubuntu22.04-py3.11
|
||||||
steps:
|
steps:
|
||||||
@@ -406,8 +394,8 @@ jobs:
|
|||||||
bash scripts/ci/npu/npu_ci_install_dependency.sh a3 diffusion
|
bash scripts/ci/npu/npu_ci_install_dependency.sh a3 diffusion
|
||||||
# copy required file from our daily cache
|
# copy required file from our daily cache
|
||||||
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
cp ~/.cache/modelscope/hub/datasets/otavia/ShareGPT_Vicuna_unfiltered/ShareGPT_V3_unfiltered_cleaned_split.json /tmp
|
||||||
# copy download through proxy
|
# copy gsm8k dataset
|
||||||
curl -o /tmp/test.jsonl -L https://gh-proxy.test.osinfra.cn/https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl
|
cp ~/.cache/modelscope/hub/datasets/tmp/test.jsonl /tmp
|
||||||
|
|
||||||
- name: Run test
|
- name: Run test
|
||||||
timeout-minutes: 60
|
timeout-minutes: 60
|
||||||
@@ -421,3 +409,45 @@ jobs:
|
|||||||
run: |
|
run: |
|
||||||
cd python
|
cd python
|
||||||
python3 sglang/multimodal_gen/test/run_suite.py --suite 8-npu
|
python3 sglang/multimodal_gen/test/run_suite.py --suite 8-npu
|
||||||
|
|
||||||
|
pr-test-finish:
|
||||||
|
needs:
|
||||||
|
[
|
||||||
|
check-changes,
|
||||||
|
|
||||||
|
stage-b-test-1-npu-a2,
|
||||||
|
stage-b-test-2-npu-a2,
|
||||||
|
stage-b-test-4-npu-a3,
|
||||||
|
stage-b-test-16-npu-a3,
|
||||||
|
|
||||||
|
multimodal-gen-test-1-npu-a3,
|
||||||
|
multimodal-gen-test-2-npu-a3,
|
||||||
|
multimodal-gen-test-8-npu-a3,
|
||||||
|
]
|
||||||
|
if: always()
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Check all dependent job statuses
|
||||||
|
run: |
|
||||||
|
# Convert the 'needs' context to a JSON string
|
||||||
|
json_needs='${{ toJson(needs) }}'
|
||||||
|
|
||||||
|
# Get a list of all job names from the JSON keys
|
||||||
|
job_names=$(echo "$json_needs" | jq -r 'keys_unsorted[]')
|
||||||
|
|
||||||
|
for job in $job_names; do
|
||||||
|
# For each job, extract its result
|
||||||
|
result=$(echo "$json_needs" | jq -r --arg j "$job" '.[$j].result')
|
||||||
|
|
||||||
|
# Print the job name and its result
|
||||||
|
echo "$job: $result"
|
||||||
|
|
||||||
|
# Check for failure or cancellation and exit if found
|
||||||
|
if [[ "$result" == "failure" || "$result" == "cancelled" ]]; then
|
||||||
|
echo "The above jobs failed."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
# If the loop completes, all jobs were successful
|
||||||
|
echo "All jobs completed successfully"
|
||||||
|
exit 0
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ on:
|
|||||||
- 'docker/npu.Dockerfile'
|
- 'docker/npu.Dockerfile'
|
||||||
workflow_dispatch:
|
workflow_dispatch:
|
||||||
schedule:
|
schedule:
|
||||||
- cron: "0 0 * * *"
|
- cron: "0 16 * * *" # Execute at 0:00 a.m. Beijing Time every day
|
||||||
|
|
||||||
concurrency:
|
concurrency:
|
||||||
group: ${{ github.workflow }}-${{ github.sha }}
|
group: ${{ github.workflow }}-${{ github.sha }}
|
||||||
|
|||||||
@@ -26,6 +26,8 @@ dependencies = [
|
|||||||
"einops",
|
"einops",
|
||||||
"fastapi",
|
"fastapi",
|
||||||
"gguf",
|
"gguf",
|
||||||
|
"hf_transfer",
|
||||||
|
"huggingface_hub",
|
||||||
"interegular",
|
"interegular",
|
||||||
"llguidance>=0.7.11,<0.8.0",
|
"llguidance>=0.7.11,<0.8.0",
|
||||||
"modelscope",
|
"modelscope",
|
||||||
|
|||||||
@@ -117,9 +117,18 @@ QWEN3_1_7B_GPTQ_INT8_WEIGHTS_PATH = os.path.join(
|
|||||||
QWEN3_235B_A22B_W8A8_WEIGHTS_PATH = os.path.join(
|
QWEN3_235B_A22B_W8A8_WEIGHTS_PATH = os.path.join(
|
||||||
MODEL_WEIGHTS_DIR, "vllm-ascend/Qwen3-235B-A22B-W8A8"
|
MODEL_WEIGHTS_DIR, "vllm-ascend/Qwen3-235B-A22B-W8A8"
|
||||||
)
|
)
|
||||||
|
QWEN3_30B_A3B_GPTQ_2507_INT4_WEIGHTS_PATH = os.path.join(
|
||||||
|
MODEL_WEIGHTS_DIR, "Qwen/Qwen3-30B-A3B-GPTQ-Int4"
|
||||||
|
)
|
||||||
|
QWEN3_30B_A3B_INSTRUCT_2507_INT4_AUTOROUND_WEIGHTS_PATH = os.path.join(
|
||||||
|
MODEL_WEIGHTS_DIR, "Intel/Qwen3-30B-A3B-Instruct-2507-int4-AutoRound"
|
||||||
|
)
|
||||||
QWEN3_30B_A3B_INSTRUCT_2507_WEIGHTS_PATH = os.path.join(
|
QWEN3_30B_A3B_INSTRUCT_2507_WEIGHTS_PATH = os.path.join(
|
||||||
MODEL_WEIGHTS_DIR, "Qwen/Qwen3-30B-A3B-Instruct-2507"
|
MODEL_WEIGHTS_DIR, "Qwen/Qwen3-30B-A3B-Instruct-2507"
|
||||||
)
|
)
|
||||||
|
QWEN3_8B_INT4_AUTOROUND_WEIGHTS_PATH = os.path.join(
|
||||||
|
MODEL_WEIGHTS_DIR, "Intel/Qwen3-8B-int4-AutoRound"
|
||||||
|
)
|
||||||
QWEN3_8B_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "Qwen/Qwen3-8B")
|
QWEN3_8B_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "Qwen/Qwen3-8B")
|
||||||
QWEN3_8B_EAGLE3_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "Qwen/Qwen3-8B_eagle3")
|
QWEN3_8B_EAGLE3_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "Qwen/Qwen3-8B_eagle3")
|
||||||
QWEN3_32B_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "Qwen/Qwen3-32B")
|
QWEN3_32B_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "Qwen/Qwen3-32B")
|
||||||
@@ -133,15 +142,6 @@ QWEN3_32B_EAGLE3_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "Qwen/Qwen3-32B-
|
|||||||
QWEN3_32B_W8A8_MINDIE_WEIGHTS_PATH = os.path.join(
|
QWEN3_32B_W8A8_MINDIE_WEIGHTS_PATH = os.path.join(
|
||||||
MODEL_WEIGHTS_DIR, "aleoyang/Qwen3-32B-w8a8-MindIE"
|
MODEL_WEIGHTS_DIR, "aleoyang/Qwen3-32B-w8a8-MindIE"
|
||||||
)
|
)
|
||||||
QWEN3_235B_A22B_W8A8_WEIGHTS_PATH = os.path.join(
|
|
||||||
MODEL_WEIGHTS_DIR, "vllm-ascend/Qwen3-235B-A22B-W8A8"
|
|
||||||
)
|
|
||||||
QWEN3_CODER_480B_A35B_INSTRUCT_W8A8_QUAROT_WEIGHTS_PATH = os.path.join(
|
|
||||||
MODEL_WEIGHTS_DIR, "Qwen3-Coder-480B-A35B-Instruct-w8a8-QuaRot"
|
|
||||||
)
|
|
||||||
QWEN3_NEXT_80B_A3B_INSTRUCT_WEIGHTS_PATH = os.path.join(
|
|
||||||
MODEL_WEIGHTS_DIR, "Qwen/Qwen3-Next-80B-A3B-Instruct"
|
|
||||||
)
|
|
||||||
QWQ_32B_W8A8_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "vllm-ascend/QWQ-32B-W8A8")
|
QWQ_32B_W8A8_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "vllm-ascend/QWQ-32B-W8A8")
|
||||||
SMOLLM_1_7B_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "HuggingFaceTB/SmolLM-1.7B")
|
SMOLLM_1_7B_WEIGHTS_PATH = os.path.join(MODEL_WEIGHTS_DIR, "HuggingFaceTB/SmolLM-1.7B")
|
||||||
STABLELM_2_1_6B_WEIGHTS_PATH = os.path.join(
|
STABLELM_2_1_6B_WEIGHTS_PATH = os.path.join(
|
||||||
|
|||||||
+4
@@ -3,6 +3,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -11,6 +12,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-1-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
||||||
"accuracy": 0.85,
|
"accuracy": 0.85,
|
||||||
+4
@@ -3,6 +3,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -11,6 +12,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-4-npu-a3", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-4-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V2-Lite-W8A8": {
|
"/root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V2-Lite-W8A8": {
|
||||||
"accuracy": 0.34,
|
"accuracy": 0.34,
|
||||||
+4
@@ -4,6 +4,7 @@ from types import SimpleNamespace
|
|||||||
import requests
|
import requests
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.run_eval import run_eval
|
from sglang.test.run_eval import run_eval
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -12,6 +13,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-1-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestAscendSamplingBackend(CustomTestCase):
|
class TestAscendSamplingBackend(CustomTestCase):
|
||||||
@classmethod
|
@classmethod
|
||||||
+4
@@ -3,6 +3,7 @@ import unittest
|
|||||||
from types import SimpleNamespace
|
from types import SimpleNamespace
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.send_one import BenchArgs, send_one_prompt
|
from sglang.test.send_one import BenchArgs, send_one_prompt
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
@@ -14,6 +15,9 @@ from sglang.test.test_utils import (
|
|||||||
write_github_step_summary,
|
write_github_step_summary,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-4-npu-a3", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestLLaDA2Mini(CustomTestCase):
|
class TestLLaDA2Mini(CustomTestCase):
|
||||||
@classmethod
|
@classmethod
|
||||||
+4
@@ -4,6 +4,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -12,6 +13,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-1-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
||||||
"accuracy": 0.84,
|
"accuracy": 0.84,
|
||||||
+4
@@ -3,6 +3,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -11,6 +12,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-1-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
||||||
"accuracy": 0.85,
|
"accuracy": 0.85,
|
||||||
+4
@@ -3,6 +3,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -11,6 +12,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-2-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-2-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
||||||
"accuracy": 0.85,
|
"accuracy": 0.85,
|
||||||
+4
@@ -2,6 +2,7 @@ import unittest
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -12,6 +13,9 @@ from sglang.test.test_utils import (
|
|||||||
run_bench_one_batch,
|
run_bench_one_batch,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-1-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
MODEL = "/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct"
|
MODEL = "/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct"
|
||||||
GSM8K_EXP_ACCURACY = 0.84
|
GSM8K_EXP_ACCURACY = 0.84
|
||||||
EXP_PREFILL_LATENCY = 0.045
|
EXP_PREFILL_LATENCY = 0.045
|
||||||
+4
@@ -4,6 +4,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_URL_FOR_TEST,
|
DEFAULT_URL_FOR_TEST,
|
||||||
@@ -11,6 +12,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-16-npu-a3", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-16-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-R1-0528-W8A8": {
|
"/root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-R1-0528-W8A8": {
|
||||||
"accuracy": 0.95,
|
"accuracy": 0.95,
|
||||||
+6
-1
@@ -4,6 +4,8 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ascend.test_ascend_utils import QWEN3_8B_INT4_AUTOROUND_WEIGHTS_PATH
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -12,10 +14,13 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-1-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/Intel/Qwen3-8B-int4-AutoRound": {
|
QWEN3_8B_INT4_AUTOROUND_WEIGHTS_PATH: {
|
||||||
"accuracy": 0.85,
|
"accuracy": 0.85,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
+8
-1
@@ -4,6 +4,10 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ascend.test_ascend_utils import (
|
||||||
|
QWEN3_30B_A3B_INSTRUCT_2507_INT4_AUTOROUND_WEIGHTS_PATH,
|
||||||
|
)
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -12,10 +16,13 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-1-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/Intel/Qwen3-30B-A3B-Instruct-2507-int4-AutoRound": {
|
QWEN3_30B_A3B_INSTRUCT_2507_INT4_AUTOROUND_WEIGHTS_PATH: {
|
||||||
"accuracy": 0.85,
|
"accuracy": 0.85,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
+8
-1
@@ -4,6 +4,10 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ascend.test_ascend_utils import (
|
||||||
|
QWEN3_30B_A3B_GPTQ_2507_INT4_WEIGHTS_PATH,
|
||||||
|
)
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -12,10 +16,13 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-1-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/Qwen/Qwen3-30B-A3B-GPTQ-Int4": {
|
QWEN3_30B_A3B_GPTQ_2507_INT4_WEIGHTS_PATH: {
|
||||||
"accuracy": 0.85,
|
"accuracy": 0.85,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
+4
@@ -12,6 +12,7 @@ from urllib.parse import urlparse
|
|||||||
import requests
|
import requests
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval
|
from sglang.test.few_shot_gsm8k import run_eval
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -21,6 +22,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-4-npu-a3", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-4-npu-a3", nightly=True)
|
||||||
|
|
||||||
if "ASCEND_RT_VISIBLE_DEVICES" not in os.environ:
|
if "ASCEND_RT_VISIBLE_DEVICES" not in os.environ:
|
||||||
os.environ["ASCEND_RT_VISIBLE_DEVICES"] = "0,1,2,3"
|
os.environ["ASCEND_RT_VISIBLE_DEVICES"] = "0,1,2,3"
|
||||||
DEFAULT_PORT_FOR_SRT_TEST_RUNNER = (
|
DEFAULT_PORT_FOR_SRT_TEST_RUNNER = (
|
||||||
+4
@@ -12,6 +12,7 @@ from urllib.parse import urlparse
|
|||||||
import requests
|
import requests
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval
|
from sglang.test.few_shot_gsm8k import run_eval
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -21,6 +22,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-1-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
if "ASCEND_RT_VISIBLE_DEVICES" not in os.environ:
|
if "ASCEND_RT_VISIBLE_DEVICES" not in os.environ:
|
||||||
os.environ["ASCEND_RT_VISIBLE_DEVICES"] = "0,1"
|
os.environ["ASCEND_RT_VISIBLE_DEVICES"] = "0,1"
|
||||||
DEFAULT_PORT_FOR_SRT_TEST_RUNNER = (
|
DEFAULT_PORT_FOR_SRT_TEST_RUNNER = (
|
||||||
+4
@@ -4,6 +4,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -12,6 +13,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-2-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-2-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V2-Lite-W8A8": {
|
"/root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V2-Lite-W8A8": {
|
||||||
"accuracy": 0.34,
|
"accuracy": 0.34,
|
||||||
+4
@@ -3,6 +3,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -11,6 +12,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-4-npu-a3", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-4-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V2-Lite-W8A8": {
|
"/root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V2-Lite-W8A8": {
|
||||||
"accuracy": 0.34,
|
"accuracy": 0.34,
|
||||||
+4
@@ -3,6 +3,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -11,6 +12,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-1-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
||||||
"accuracy": 0.84,
|
"accuracy": 0.84,
|
||||||
+4
@@ -3,6 +3,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -11,6 +12,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-2-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-2-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
||||||
"accuracy": 0.85,
|
"accuracy": 0.85,
|
||||||
+4
@@ -4,6 +4,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
@@ -12,6 +13,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-2-npu-a2", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-2-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
"/root/.cache/modelscope/hub/models/Qwen/Qwen2.5-7B-Instruct": {
|
||||||
"accuracy": 0.85,
|
"accuracy": 0.85,
|
||||||
+4
@@ -3,6 +3,7 @@ from types import SimpleNamespace
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from sglang.srt.utils import kill_process_tree
|
from sglang.srt.utils import kill_process_tree
|
||||||
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
DEFAULT_URL_FOR_TEST,
|
DEFAULT_URL_FOR_TEST,
|
||||||
@@ -10,6 +11,9 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
register_npu_ci(est_time=400, suite="stage-b-test-4-npu-a3", nightly=False)
|
||||||
|
register_npu_ci(est_time=400, suite="nightly-4-npu-a3", nightly=True)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
"Qwen/Qwen3-30B-A3B-Instruct-2507": {
|
"Qwen/Qwen3-30B-A3B-Instruct-2507": {
|
||||||
"accuracy": 0.90,
|
"accuracy": 0.90,
|
||||||
@@ -11,7 +11,7 @@ from sglang.test.test_utils import CustomTestCase, get_similarities
|
|||||||
|
|
||||||
register_npu_ci(
|
register_npu_ci(
|
||||||
est_time=400,
|
est_time=400,
|
||||||
suite="nightly-1-npu-a3",
|
suite="full-1-npu-a3",
|
||||||
nightly=True,
|
nightly=True,
|
||||||
disabled="embeddings are not all close",
|
disabled="embeddings are not all close",
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ from sglang.test.ascend.test_ascend_utils import AFM_4_5B_BASE_WEIGHTS_PATH
|
|||||||
from sglang.test.ci.ci_register import register_npu_ci
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
register_npu_ci(est_time=400, suite="full-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestAFM(GSM8KAscendMixin, CustomTestCase):
|
class TestAFM(GSM8KAscendMixin, CustomTestCase):
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ from sglang.test.ascend.test_ascend_utils import (
|
|||||||
from sglang.test.ci.ci_register import register_npu_ci
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-2-npu-a3", nightly=False)
|
register_npu_ci(est_time=400, suite="full-2-npu-a3", nightly=False)
|
||||||
|
|
||||||
|
|
||||||
class TestC4AI(GSM8KAscendMixin, CustomTestCase):
|
class TestC4AI(GSM8KAscendMixin, CustomTestCase):
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ from sglang.test.ascend.test_ascend_utils import EXAONE_3_5_7_8B_INSTRUCT_WEIGHT
|
|||||||
from sglang.test.ci.ci_register import register_npu_ci
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=False)
|
register_npu_ci(est_time=400, suite="full-1-npu-a3", nightly=False)
|
||||||
|
|
||||||
|
|
||||||
class TestEXAONE(GSM8KAscendMixin, CustomTestCase):
|
class TestEXAONE(GSM8KAscendMixin, CustomTestCase):
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ from sglang.test.ascend.test_ascend_utils import (
|
|||||||
from sglang.test.ci.ci_register import register_npu_ci
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
register_npu_ci(est_time=400, suite="full-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestGranite(GSM8KAscendMixin, CustomTestCase):
|
class TestGranite(GSM8KAscendMixin, CustomTestCase):
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ from sglang.test.ascend.test_ascend_utils import GRANITE_3_1_8B_INSTRUCT_WEIGHTS
|
|||||||
from sglang.test.ci.ci_register import register_npu_ci
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
register_npu_ci(est_time=400, suite="full-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestGranite(GSM8KAscendMixin, CustomTestCase):
|
class TestGranite(GSM8KAscendMixin, CustomTestCase):
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ from sglang.test.test_utils import CustomTestCase
|
|||||||
|
|
||||||
register_npu_ci(
|
register_npu_ci(
|
||||||
est_time=400,
|
est_time=400,
|
||||||
suite="nightly-16-npu-a3",
|
suite="full-16-npu-a3",
|
||||||
nightly=False,
|
nightly=False,
|
||||||
disabled="https://github.com/Ascend/sglang/issues/25",
|
disabled="https://github.com/Ascend/sglang/issues/25",
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ from sglang.test.ascend.test_ascend_utils import LING_LITE_WEIGHTS_PATH
|
|||||||
from sglang.test.ci.ci_register import register_npu_ci
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-2-npu-a3", nightly=True)
|
register_npu_ci(est_time=400, suite="full-2-npu-a3", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestLingLite(GSM8KAscendMixin, CustomTestCase):
|
class TestLingLite(GSM8KAscendMixin, CustomTestCase):
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ from sglang.test.ascend.test_ascend_utils import MIMO_7B_RL_WEIGHTS_PATH
|
|||||||
from sglang.test.ci.ci_register import register_npu_ci
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
register_npu_ci(est_time=400, suite="full-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestMiMo7BRL(GSM8KAscendMixin, CustomTestCase):
|
class TestMiMo7BRL(GSM8KAscendMixin, CustomTestCase):
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ from sglang.test.test_utils import CustomTestCase
|
|||||||
|
|
||||||
register_npu_ci(
|
register_npu_ci(
|
||||||
est_time=400,
|
est_time=400,
|
||||||
suite="nightly-1-npu-a3",
|
suite="full-1-npu-a3",
|
||||||
nightly=False,
|
nightly=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ from sglang.test.ascend.test_ascend_utils import SMOLLM_1_7B_WEIGHTS_PATH
|
|||||||
from sglang.test.ci.ci_register import register_npu_ci
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
register_npu_ci(est_time=400, suite="full-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestSmolLM(GSM8KAscendMixin, CustomTestCase):
|
class TestSmolLM(GSM8KAscendMixin, CustomTestCase):
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ from sglang.test.ascend.test_ascend_utils import STABLELM_2_1_6B_WEIGHTS_PATH
|
|||||||
from sglang.test.ci.ci_register import register_npu_ci
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
register_npu_ci(est_time=400, suite="full-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestStablelm(GSM8KAscendMixin, CustomTestCase):
|
class TestStablelm(GSM8KAscendMixin, CustomTestCase):
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ from sglang.test.runners import HFRunner, SRTRunner
|
|||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=True)
|
register_npu_ci(est_time=400, suite="full-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
MODELS = [
|
MODELS = [
|
||||||
(
|
(
|
||||||
|
|||||||
@@ -13,8 +13,8 @@ from sglang.test.test_utils import CustomTestCase
|
|||||||
|
|
||||||
register_npu_ci(
|
register_npu_ci(
|
||||||
est_time=400,
|
est_time=400,
|
||||||
suite="nightly-4-npu-a3",
|
suite="full-4-npu-a3",
|
||||||
nightly=False,
|
nightly=True,
|
||||||
)
|
)
|
||||||
|
|
||||||
PROMPT = (
|
PROMPT = (
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ from sglang.test.ci.ci_register import register_npu_ci
|
|||||||
from sglang.test.runners import HFRunner, SRTRunner
|
from sglang.test.runners import HFRunner, SRTRunner
|
||||||
from sglang.test.test_utils import CustomTestCase
|
from sglang.test.test_utils import CustomTestCase
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-1-npu-a3", nightly=False)
|
register_npu_ci(est_time=400, suite="full-1-npu-a3", nightly=True)
|
||||||
|
|
||||||
MODELS = [
|
MODELS = [
|
||||||
(SKYWORK_REWARD_LLAMA_3_1_8B_V0_2_WEIGHTS_PATH, 1, 4e-2),
|
(SKYWORK_REWARD_LLAMA_3_1_8B_V0_2_WEIGHTS_PATH, 1, 4e-2),
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ from sglang.test.ascend.test_ascend_utils import QWEN3_VL_4B_INSTRUCT_WEIGHTS_PA
|
|||||||
from sglang.test.ascend.vlm_utils import TestVLMModels
|
from sglang.test.ascend.vlm_utils import TestVLMModels
|
||||||
from sglang.test.ci.ci_register import register_npu_ci
|
from sglang.test.ci.ci_register import register_npu_ci
|
||||||
|
|
||||||
register_npu_ci(est_time=400, suite="nightly-4-npu-a3", nightly=True)
|
register_npu_ci(est_time=400, suite="full-4-npu-a3", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestQwen3VL4B(TestVLMModels):
|
class TestQwen3VL4B(TestVLMModels):
|
||||||
|
|||||||
@@ -105,6 +105,11 @@ NIGHTLY_SUITES = {
|
|||||||
"nightly-4-npu-a3",
|
"nightly-4-npu-a3",
|
||||||
"nightly-8-npu-a3",
|
"nightly-8-npu-a3",
|
||||||
"nightly-16-npu-a3",
|
"nightly-16-npu-a3",
|
||||||
|
"full-1-npu-a3",
|
||||||
|
"full-2-npu-a3",
|
||||||
|
"full-4-npu-a3",
|
||||||
|
"full-8-npu-a3",
|
||||||
|
"full-16-npu-a3",
|
||||||
],
|
],
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user