From 4588d1f21505a0bc29d1420767f8607abca45198 Mon Sep 17 00:00:00 2001 From: Li Wang Date: Tue, 9 Dec 2025 15:45:14 +0800 Subject: [PATCH] [CI] Use arm node for unit tests (#4819) ### What this PR does / why we need it? Use arm node for unit tests Signed-off-by: wangli --- .github/workflows/vllm_ascend_test.yaml | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/.github/workflows/vllm_ascend_test.yaml b/.github/workflows/vllm_ascend_test.yaml index 013fadf..07c258d 100644 --- a/.github/workflows/vllm_ascend_test.yaml +++ b/.github/workflows/vllm_ascend_test.yaml @@ -75,7 +75,7 @@ jobs: name: unit test # only trigger unit test after lint passed and the change is e2e and ut related. if: ${{ needs.lint.result == 'success' && (needs.changes.outputs.e2e_tracker == 'true' || needs.changes.outputs.ut_tracker == 'true') }} - runs-on: ubuntu-latest + runs-on: ubuntu-22.04-arm container: image: quay.io/ascend/cann:8.2.rc1-910b-ubuntu22.04-py3.11 env: @@ -100,7 +100,7 @@ jobs: - name: Install vllm-project/vllm from source working-directory: ./vllm-empty run: | - VLLM_TARGET_DEVICE=empty python3 -m pip install . --extra-index https://download.pytorch.org/whl/cpu/ + VLLM_TARGET_DEVICE=empty python3 -m pip install . python3 -m pip uninstall -y triton - name: Checkout vllm-project/vllm-ascend repo @@ -109,16 +109,16 @@ jobs: - name: Install vllm-project/vllm-ascend run: | export PIP_EXTRA_INDEX_URL=https://mirrors.huaweicloud.com/ascend/repos/pypi - export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/Ascend/ascend-toolkit/latest/x86_64-linux/devlib - python3 -m pip install -r requirements-dev.txt --extra-index https://download.pytorch.org/whl/cpu/ - python3 -m pip install -v . --extra-index https://download.pytorch.org/whl/cpu/ + export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/Ascend/ascend-toolkit/latest/arm64-linux/devlib + python3 -m pip install -r requirements-dev.txt + python3 -m pip install -v . - name: Run unit test env: VLLM_WORKER_MULTIPROC_METHOD: spawn TORCH_DEVICE_BACKEND_AUTOLOAD: 0 run: | - export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/Ascend/ascend-toolkit/latest/x86_64-linux/devlib + export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/Ascend/ascend-toolkit/latest/arm64-linux/devlib pytest -sv --cov --cov-report=xml:unittests-coverage.xml tests/ut \ --ignore tests/ut/attention/test_attention_v1.py - name: Upload coverage to Codecov