Skip to content

[BACKEND] Update ascend backend to 29d243e 20260129 #299

[BACKEND] Update ascend backend to 29d243e 20260129

[BACKEND] Update ascend backend to 29d243e 20260129 #299

name: Ascend-Build-And-Test
on:
push:
branches: [ "triton_v3.2.x" ]
pull_request:
branches: [ "triton_v3.2.x" ]
concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
cancel-in-progress: true
jobs:
ascend-build-and-test:
runs-on: flagtree-ascend
if: ${{ github.repository == 'FlagTree/flagtree' || github.repository == 'flagos-ai/flagtree' }}
steps:
- name: Setup environment
shell: bash
run: |
source ~/env.sh
env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true
- name: Checkout code (attempt 1)
id: checkout1
uses: actions/checkout@v6
with:
fetch-depth: 0
continue-on-error: true
- name: Sleep before checkout2
if: steps.checkout1.outcome == 'failure'
run: |
echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
sleep 120
- name: Checkout code (attempt 2)
id: checkout2
if: steps.checkout1.outcome == 'failure'
uses: actions/checkout@v6
with:
fetch-depth: 0
continue-on-error: true
- name: Sleep before final checkout
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
run: |
echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
sleep 180
- name: Checkout code (final attempt)
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
uses: actions/checkout@v6
with:
fetch-depth: 0
- name: Verify checkout success
if: success()
run: echo "Checkout completed successfully"
- name: Check if only docs files changed
id: check_files
uses: ./.github/actions/check-docs-only
- name: FlagTree Build on Ascend
if: steps.check_files.outputs.only_docs_changed != 'true'
shell: bash
run: |
set -x
export FLAGTREE_BACKEND=ascend
source ~/env.sh
cd python
MAX_JOBS=32 python3 -m pip install . --no-build-isolation -vvv
- name: FlagTree Test on Ascend
if: steps.check_files.outputs.only_docs_changed != 'true'
shell: bash
run: |
set -x
source /usr/local/Ascend/ascend-toolkit/set_env.sh
# tutorials
pushd third_party/ascend/tutorials
python3 01-vector-add.py
python3 02-fused-softmax.py
python3 03-layer-norm.py
python3 04-fused-attention.py
python3 06-demo-autotune.py
python3 07-profiler.py
python3 08-demo-libentry.py
python3 09-gather.py >output-09-gather.log
python3 10-gather_sorted.py >output-10-gather_sorted.log
#python3 11-rab_time.py
#python3 12-hstu_attention.py
python3 13-matrix-multiplication-optimized.py
python3 14-accuracy-comparison.py
#python3 15-embedding_gather_demo.py
popd
# pytest_ut
pushd third_party/ascend/unittest/pytest_ut
python3 -m pytest . \
--ignore=test_add_mindspore.py \
--ignore=test_softmax_mindspore.py \
--ignore=test_assume.py \
--ignore=test_custom.py \
--ignore=test_dot_scaled.py \
--ignore=test_elementwise_ceil.py \
--ignore=test_elementwise_clip.py \
--ignore=test_elementwise_f2i.py \
--ignore=test_elementwise_floor.py \
--ignore=test_elementwise_i2f.py \
--ignore=test_elementwise_round.py \
--ignore=test_custom.py \
--ignore=test_custom.py \
--ignore=test_custom.py \
--ignore=test_custom.py \
#--ignore=test_index_select.py \
#--ignore=test_linearize_permute.py \
#--ignore=test_logical_and.py \
#--ignore=test_logical_or.py \
#--ignore=test_triton_unified_attention.py
popd