[CI]Add global env to runner (#537)
### What this PR does / why we need it? - add `HF_TOKEN` as global var to the runner - add `HF_ENDPOINT` as global var to the runner - change concurrency group, rely on current pr num --------- Signed-off-by: wangli <wangli858794774@gmail.com>
This commit is contained in:
15
.github/workflows/vllm_ascend_test.yaml
vendored
15
.github/workflows/vllm_ascend_test.yaml
vendored
@@ -48,13 +48,21 @@ jobs:
|
|||||||
os: [linux-arm64-npu-1, linux-arm64-npu-4]
|
os: [linux-arm64-npu-1, linux-arm64-npu-4]
|
||||||
vllm_verison: [main, v0.8.4]
|
vllm_verison: [main, v0.8.4]
|
||||||
concurrency:
|
concurrency:
|
||||||
group: |
|
group: >
|
||||||
${{ matrix.os == 'linux-arm64-npu-4' && 'limit-npu-4' || format('job-{0}-{1}', matrix.os, matrix.vllm_verison) }}
|
${{
|
||||||
|
matrix.os == 'linux-arm64-npu-4'
|
||||||
|
&& github.event.pull_request.number
|
||||||
|
&& format('pr-{0}-limit-npu-4', github.event.pull_request.number)
|
||||||
|
|| format('job-{0}-{1}-{2}', matrix.os, matrix.vllm_verison, github.event.pull_request.number)
|
||||||
|
}}
|
||||||
cancel-in-progress: false
|
cancel-in-progress: false
|
||||||
name: vLLM Ascend test
|
name: vLLM Ascend test
|
||||||
runs-on: ${{ matrix.os }}
|
runs-on: ${{ matrix.os }}
|
||||||
container:
|
container:
|
||||||
image: quay.io/ascend/cann:8.0.0-910b-ubuntu22.04-py3.10
|
image: quay.io/ascend/cann:8.0.0-910b-ubuntu22.04-py3.10
|
||||||
|
env:
|
||||||
|
HF_ENDPOINT: https://hf-mirror.com
|
||||||
|
HF_TOKEN: ${{ secrets.HF_TOKEN }}
|
||||||
steps:
|
steps:
|
||||||
- name: Check npu and CANN info
|
- name: Check npu and CANN info
|
||||||
run: |
|
run: |
|
||||||
@@ -112,7 +120,6 @@ jobs:
|
|||||||
- name: Run vllm-project/vllm-ascend test on V0 engine
|
- name: Run vllm-project/vllm-ascend test on V0 engine
|
||||||
env:
|
env:
|
||||||
VLLM_USE_V1: 0
|
VLLM_USE_V1: 0
|
||||||
HF_ENDPOINT: https://hf-mirror.com
|
|
||||||
run: |
|
run: |
|
||||||
if [[ "${{ matrix.os }}" == "linux-arm64-npu-1" ]]; then
|
if [[ "${{ matrix.os }}" == "linux-arm64-npu-1" ]]; then
|
||||||
pytest -sv tests/singlecard
|
pytest -sv tests/singlecard
|
||||||
@@ -126,7 +133,6 @@ jobs:
|
|||||||
env:
|
env:
|
||||||
VLLM_USE_V1: 1
|
VLLM_USE_V1: 1
|
||||||
VLLM_WORKER_MULTIPROC_METHOD: spawn
|
VLLM_WORKER_MULTIPROC_METHOD: spawn
|
||||||
HF_ENDPOINT: https://hf-mirror.com
|
|
||||||
run: |
|
run: |
|
||||||
if [[ "${{ matrix.os }}" == "linux-arm64-npu-1" ]]; then
|
if [[ "${{ matrix.os }}" == "linux-arm64-npu-1" ]]; then
|
||||||
pytest -sv tests/singlecard
|
pytest -sv tests/singlecard
|
||||||
@@ -140,6 +146,5 @@ jobs:
|
|||||||
env:
|
env:
|
||||||
VLLM_USE_V1: 0
|
VLLM_USE_V1: 0
|
||||||
PYTORCH_NPU_ALLOC_CONF: max_split_size_mb:256
|
PYTORCH_NPU_ALLOC_CONF: max_split_size_mb:256
|
||||||
HF_ENDPOINT: https://hf-mirror.com
|
|
||||||
run: |
|
run: |
|
||||||
pytest -sv
|
pytest -sv
|
||||||
|
|||||||
Reference in New Issue
Block a user