EngineX/xc-llm-kunlun
3
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

update ci workflow (#159)

Browse Source 
Signed-off-by: tanjunchen <tanjunchen20@gmail.com>
This commit is contained in:
tanjunchen
2026-01-28 20:28:38 +08:00
committed by GitHub
parent 7c2966a98c
commit 1e1e870a71
5 changed files with 185 additions and 195 deletions
Download Patch File Download Diff File Expand all files Collapse all files

88
.github/workflows/ut.yml vendored
View File

@@ -1,57 +1,53 @@
name: Unit Test
# name: Unit Test
on:
pull_request:
branches:
- main
# on:
# pull_request:
# branches:
# - main
jobs:
test-kunlun:
runs-on:
labels:
- self-hosted
- Linux
- X64
- test-1 # Actions Runner Label
# jobs:
# test-kunlun:
# runs-on:
# labels:
# - self-hosted
# - Linux
# - X64
# - test-1 # Actions Runner Label
steps:
- name: Checkout Code
uses: actions/checkout@v4
# steps:
# - name: Checkout Code
# uses: actions/checkout@v4
- name: Install vLLM-Kunlun Dependencies
run: |
pip install -r requirements.txt
# - name: Install vLLM-Kunlun Dependencies
# run: |
# pip install -r requirements.txt
python setup.py build
python setup.py develop
# python setup.py build
# python setup.py develop
# Install the KL3-customized build of PyTorch
wget -O xpytorch-cp310-torch251-ubuntu2004-x64.run https://baidu-kunlun-public.su.bcebos.com/v1/baidu-kunlun-share/1130/xpytorch-cp310-torch251-ubuntu2004-x64.run?authorization=bce-auth-v1%2FALTAKypXxBzU7gg4Mk4K4c6OYR%2F2025-12-02T05%3A01%3A27Z%2F-1%2Fhost%2Ff3cf499234f82303891aed2bcb0628918e379a21e841a3fac6bd94afef491ff7
bash xpytorch-cp310-torch251-ubuntu2004-x64.run
# # Install the KL3-customized build of PyTorch
# wget -O xpytorch-cp310-torch251-ubuntu2004-x64.run https://baidu-kunlun-public.su.bcebos.com/v1/baidu-kunlun-share/1130/xpytorch-cp310-torch251-ubuntu2004-x64.run?authorization=bce-auth-v1%2FALTAKypXxBzU7gg4Mk4K4c6OYR%2F2025-12-02T05%3A01%3A27Z%2F-1%2Fhost%2Ff3cf499234f82303891aed2bcb0628918e379a21e841a3fac6bd94afef491ff7
# bash xpytorch-cp310-torch251-ubuntu2004-x64.run
# Install custom ops
pip install "https://baidu-kunlun-public.su.bcebos.com/v1/baidu-kunlun-share/1130/xtorch_ops-0.1.2209%2B6752ad20-cp310-cp310-linux_x86_64.whl?authorization=bce-auth-v1%2FALTAKypXxBzU7gg4Mk4K4c6OYR%2F2025-12-05T06%3A18%3A00Z%2F-1%2Fhost%2F14936c2b7e7c557c1400e4c467c79f7a9217374a7aa4a046711ac4d948f460cd"
# # Install custom ops
# pip install "https://baidu-kunlun-public.su.bcebos.com/v1/baidu-kunlun-share/1130/xtorch_ops-0.1.2209%2B6752ad20-cp310-cp310-linux_x86_64.whl?authorization=bce-auth-v1%2FALTAKypXxBzU7gg4Mk4K4c6OYR%2F2025-12-05T06%3A18%3A00Z%2F-1%2Fhost%2F14936c2b7e7c557c1400e4c467c79f7a9217374a7aa4a046711ac4d948f460cd"
# Install the KLX3 custom Triton build
pip install "https://cce-ai-models.bj.bcebos.com/v1/vllm-kunlun-0.11.0/triton-3.0.0%2Bb2cde523-cp310-cp310-linux_x86_64.whl"
# # Install the KLX3 custom Triton build
# pip install "https://cce-ai-models.bj.bcebos.com/v1/vllm-kunlun-0.11.0/triton-3.0.0%2Bb2cde523-cp310-cp310-linux_x86_64.whl"
# Install the AIAK custom ops library
pip install "https://cce-ai-models.bj.bcebos.com/XSpeedGate-whl/release_merge/20251219_152418/xspeedgate_ops-0.0.0-cp310-cp310-linux_x86_64.whl"
# # Install the AIAK custom ops library
# pip install "https://cce-ai-models.bj.bcebos.com/XSpeedGate-whl/release_merge/20251219_152418/xspeedgate_ops-0.0.0-cp310-cp310-linux_x86_64.whl"
- name: Install vLLM
run: |
pip install vllm==0.11.0 --no-build-isolation --no-deps --no-deps --index-url https://pip.baidu-int.com/simple/
# - name: Install vLLM
# run: |
# pip install vllm==0.11.0 --no-build-isolation --no-deps --no-deps --index-url https://pip.baidu-int.com/simple/
- name: Install Test Dependencies
run: |
pip install pytest
- name: Run Unit Test
run: |
echo "Running full suite..."
export XPU_VISIBLE_DEVICES=1
pytest \
-vs \
--cov=vllm_kunlun \
--cov-report=term-missing \
-p no:warnings tests/ut
# - name: Run Unit Test
# run: |
# echo "Running full suite..."
# export XPU_VISIBLE_DEVICES=1
# pytest \
# -vs \
# --cov=vllm_kunlun \
# --cov-report=term-missing \
# -p no:warnings tests/ut