Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions .github/workflows/_e2e_nightly_single_node.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -123,8 +123,7 @@ jobs:
pip install custom_ops-1.0-cp311-cp311-linux_aarch64.whl
. /usr/local/Ascend/ascend-toolkit/set_env.sh

- name: Install triton-ascend
if: ${{ inputs.name == 'test_custom_op' }}
- name: Install Ascend toolkit & triton_ascend
shell: bash -l {0}
run: |
BISHENG_NAME="Ascend-BiSheng-toolkit_aarch64_20260105.run"
Expand Down
3 changes: 1 addition & 2 deletions .github/workflows/_e2e_nightly_single_node_models.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -104,8 +104,7 @@ jobs:
pip install -r requirements-dev.txt
pip install -v -e .

- name: Install Ascend toolkit & triton_ascend (for Qwen3-Next-80B-A3B-Instruct)
if: ${{ inputs.runner == 'linux-aarch64-a2-4' && contains(inputs.model_list, 'Qwen3-Next-80B-A3B-Instruct') }}
- name: Install Ascend toolkit & triton_ascend
shell: bash -l {0}
run: |
BISHENG_NAME="Ascend-BiSheng-toolkit_aarch64_20260105.run"
Expand Down
26 changes: 26 additions & 0 deletions tests/e2e/nightly/multi_node/scripts/run.sh
Original file line number Diff line number Diff line change
Expand Up @@ -125,6 +125,31 @@ install_extra_components() {
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() {
pgrep python3 | xargs -r kill -9
pgrep VLLM | xargs -r kill -9
Expand Down Expand Up @@ -152,6 +177,7 @@ main() {
check_npu_info
check_and_config
show_vllm_info
install_triton_ascend
if [[ "$CONFIG_YAML_PATH" == *"DeepSeek-V3_2-Exp-bf16.yaml" ]]; then
install_extra_components
fi
Expand Down
Loading