57 lines
2.1 KiB
YAML
57 lines
2.1 KiB
YAML
name: Unit Test
|
|
|
|
on:
|
|
pull_request:
|
|
branches:
|
|
- main
|
|
|
|
jobs:
|
|
test-kunlun:
|
|
runs-on:
|
|
labels:
|
|
- self-hosted
|
|
- Linux
|
|
- X64
|
|
- test-1 # Actions Runner Label
|
|
|
|
steps:
|
|
- name: Checkout Code
|
|
uses: actions/checkout@v4
|
|
|
|
- name: Install vLLM-Kunlun Dependencies
|
|
run: |
|
|
pip install -r requirements.txt
|
|
|
|
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 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 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 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 |