diff --git a/.github/workflows/_e2e_nightly_single_node.yaml b/.github/workflows/_e2e_nightly_single_node.yaml index 92345042..660f593c 100644 --- a/.github/workflows/_e2e_nightly_single_node.yaml +++ b/.github/workflows/_e2e_nightly_single_node.yaml @@ -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" diff --git a/.github/workflows/_e2e_nightly_single_node_models.yaml b/.github/workflows/_e2e_nightly_single_node_models.yaml index 2cb30f0e..6cb1a371 100644 --- a/.github/workflows/_e2e_nightly_single_node_models.yaml +++ b/.github/workflows/_e2e_nightly_single_node_models.yaml @@ -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" diff --git a/tests/e2e/nightly/multi_node/scripts/run.sh b/tests/e2e/nightly/multi_node/scripts/run.sh index 7bbb9e08..d6cca47e 100644 --- a/tests/e2e/nightly/multi_node/scripts/run.sh +++ b/tests/e2e/nightly/multi_node/scripts/run.sh @@ -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 @@ -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