[CI] Add triton ascend in nightly CI (#5716)
### What this PR does / why we need it?
Add triton ascend in nightly
### Does this PR introduce _any_ user-facing change?
### How was this patch tested?
- vLLM version: v0.13.0
- vLLM main:
2f4e6548ef
---------
Signed-off-by: Meihan-chen <jcccx.cmh@gmail.com>
This commit is contained in:
@@ -123,8 +123,7 @@ jobs:
|
|||||||
pip install custom_ops-1.0-cp311-cp311-linux_aarch64.whl
|
pip install custom_ops-1.0-cp311-cp311-linux_aarch64.whl
|
||||||
. /usr/local/Ascend/ascend-toolkit/set_env.sh
|
. /usr/local/Ascend/ascend-toolkit/set_env.sh
|
||||||
|
|
||||||
- name: Install triton-ascend
|
- name: Install Ascend toolkit & triton_ascend
|
||||||
if: ${{ inputs.name == 'test_custom_op' }}
|
|
||||||
shell: bash -l {0}
|
shell: bash -l {0}
|
||||||
run: |
|
run: |
|
||||||
BISHENG_NAME="Ascend-BiSheng-toolkit_aarch64_20260105.run"
|
BISHENG_NAME="Ascend-BiSheng-toolkit_aarch64_20260105.run"
|
||||||
|
|||||||
@@ -104,8 +104,7 @@ jobs:
|
|||||||
pip install -r requirements-dev.txt
|
pip install -r requirements-dev.txt
|
||||||
pip install -v -e .
|
pip install -v -e .
|
||||||
|
|
||||||
- name: Install Ascend toolkit & triton_ascend (for Qwen3-Next-80B-A3B-Instruct)
|
- name: Install Ascend toolkit & triton_ascend
|
||||||
if: ${{ inputs.runner == 'linux-aarch64-a2-4' && contains(inputs.model_list, 'Qwen3-Next-80B-A3B-Instruct') }}
|
|
||||||
shell: bash -l {0}
|
shell: bash -l {0}
|
||||||
run: |
|
run: |
|
||||||
BISHENG_NAME="Ascend-BiSheng-toolkit_aarch64_20260105.run"
|
BISHENG_NAME="Ascend-BiSheng-toolkit_aarch64_20260105.run"
|
||||||
|
|||||||
@@ -125,6 +125,31 @@ install_extra_components() {
|
|||||||
echo "====> Extra components installation completed"
|
echo "====> Extra components installation completed"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
install_triton_ascend() {
|
||||||
|
echo "====> Installing triton_ascend"
|
||||||
|
|
||||||
|
BISHENG_NAME="Ascend-BiSheng-toolkit_aarch64_20260105.run"
|
||||||
|
BISHENG_URL="https://vllm-ascend.obs.cn-north-4.myhuaweicloud.com/vllm-ascend/${BISHENG_NAME}"
|
||||||
|
|
||||||
|
if ! wget -q -O "${BISHENG_NAME}" "${BISHENG_URL}"; then
|
||||||
|
echo "Failed to download ${BISHENG_NAME}"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
chmod +x "${BISHENG_NAME}"
|
||||||
|
|
||||||
|
if ! "./${BISHENG_NAME}" --install; then
|
||||||
|
rm -f "${BISHENG_NAME}"
|
||||||
|
echo "Failed to install ${BISHENG_NAME}"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
rm -f "${BISHENG_NAME}"
|
||||||
|
|
||||||
|
export PATH=/usr/local/Ascend/tools/bishengir/bin:$PATH
|
||||||
|
which bishengir-compile
|
||||||
|
python3 -m pip install -i https://test.pypi.org/simple/ triton-ascend==3.2.0.dev20260105
|
||||||
|
echo "====> Triton ascend installation completed"
|
||||||
|
}
|
||||||
|
|
||||||
kill_npu_processes() {
|
kill_npu_processes() {
|
||||||
pgrep python3 | xargs -r kill -9
|
pgrep python3 | xargs -r kill -9
|
||||||
pgrep VLLM | xargs -r kill -9
|
pgrep VLLM | xargs -r kill -9
|
||||||
@@ -152,6 +177,7 @@ main() {
|
|||||||
check_npu_info
|
check_npu_info
|
||||||
check_and_config
|
check_and_config
|
||||||
show_vllm_info
|
show_vllm_info
|
||||||
|
install_triton_ascend
|
||||||
if [[ "$CONFIG_YAML_PATH" == *"DeepSeek-V3_2-Exp-bf16.yaml" ]]; then
|
if [[ "$CONFIG_YAML_PATH" == *"DeepSeek-V3_2-Exp-bf16.yaml" ]]; then
|
||||||
install_extra_components
|
install_extra_components
|
||||||
fi
|
fi
|
||||||
|
|||||||
Reference in New Issue
Block a user