Skip to content

Commit 9859e73

Browse files
authored
[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>
1 parent 00de2ee commit 9859e73

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

.github/workflows/vllm_ascend_test.yaml

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -48,13 +48,21 @@ jobs:
4848
os: [linux-arm64-npu-1, linux-arm64-npu-4]
4949
vllm_verison: [main, v0.8.4]
5050
concurrency:
51-
group: |
52-
${{ matrix.os == 'linux-arm64-npu-4' && 'limit-npu-4' || format('job-{0}-{1}', matrix.os, matrix.vllm_verison) }}
51+
group: >
52+
${{
53+
matrix.os == 'linux-arm64-npu-4'
54+
&& github.event.pull_request.number
55+
&& format('pr-{0}-limit-npu-4', github.event.pull_request.number)
56+
|| format('job-{0}-{1}-{2}', matrix.os, matrix.vllm_verison, github.event.pull_request.number)
57+
}}
5358
cancel-in-progress: false
5459
name: vLLM Ascend test
5560
runs-on: ${{ matrix.os }}
5661
container:
5762
image: quay.io/ascend/cann:8.0.0-910b-ubuntu22.04-py3.10
63+
env:
64+
HF_ENDPOINT: https://hf-mirror.com
65+
HF_TOKEN: ${{ secrets.HF_TOKEN }}
5866
steps:
5967
- name: Check npu and CANN info
6068
run: |
@@ -112,7 +120,6 @@ jobs:
112120
- name: Run vllm-project/vllm-ascend test on V0 engine
113121
env:
114122
VLLM_USE_V1: 0
115-
HF_ENDPOINT: https://hf-mirror.com
116123
run: |
117124
if [[ "${{ matrix.os }}" == "linux-arm64-npu-1" ]]; then
118125
pytest -sv tests/singlecard
@@ -126,7 +133,6 @@ jobs:
126133
env:
127134
VLLM_USE_V1: 1
128135
VLLM_WORKER_MULTIPROC_METHOD: spawn
129-
HF_ENDPOINT: https://hf-mirror.com
130136
run: |
131137
if [[ "${{ matrix.os }}" == "linux-arm64-npu-1" ]]; then
132138
pytest -sv tests/singlecard
@@ -140,6 +146,5 @@ jobs:
140146
env:
141147
VLLM_USE_V1: 0
142148
PYTORCH_NPU_ALLOC_CONF: max_split_size_mb:256
143-
HF_ENDPOINT: https://hf-mirror.com
144149
run: |
145150
pytest -sv

0 commit comments

Comments
 (0)