[HINT] Add Triton v3.2.x hint manager #277
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| 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 | |
| pip uninstall -y triton | |
| export FLAGTREE_BACKEND=ascend | |
| source ~/env.sh | |
| cd python | |
| MAX_JOBS=32 python3 -m pip install . --no-build-isolation | |
| - 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 | |
| python3 third_party/ascend/examples/tutorials/01-vector-add.py | |
| python3 third_party/ascend/examples/tutorials/02-fused-softmax.py | |
| python3 third_party/ascend/examples/tutorials/03-layer-norm.py | |
| python3 third_party/ascend/examples/tutorials/04-fused-attention.py | |
| python3 third_party/ascend/examples/tutorials/06-demo-autotune.py | |
| python3 third_party/ascend/examples/tutorials/07-profiler.py | |
| python3 third_party/ascend/examples/tutorials/09-gather.py | |
| python3 third_party/ascend/examples/tutorials/10-gather_sorted.py | |
| python3 third_party/ascend/examples/tutorials/11-rab_time.py | |
| python3 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized.py | |
| python3 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized-flagtree.py | |
| python3 third_party/ascend/examples/tutorials/14-accuracy-comparison.py | |
| python3 python/test/ops/01_vector_add/01_vector_add.py | |
| python3 python/test/ops/abs/abs.py | |
| python3 python/test/ops/addmm/addmm.py | |
| python3 python/test/ops/addmm/addmm_ascend.py | |
| python3 python/test/ops/amax/amax.py | |
| python3 python/test/ops/amax/amax_ascend_perf.py | |
| python3 python/test/ops/apply_rotary_pos_emb/apply_rotary_pos_emb.py | |
| python3 python/test/ops/apply_rotary_pos_emb/apply_rotary_pos_emb_ascend.py | |
| python3 python/test/ops/argmin/argmin.py | |
| python3 python/test/ops/argmin/argmin_ascend_perf.py | |
| python3 python/test/ops/bmm/bmm_ascend.py | |
| python3 python/test/ops/cumsum/cumsum.py | |
| python3 python/test/ops/min_dim/min_dim.py | |
| python3 python/test/ops/min_dim/min_dim_ascend_perf.py | |
| python3 python/test/ops/sum_dim/sum_dim.py | |
| python3 python/test/ops/varmean/var_mean_ascend.py | |
| python3 -m pytest third_party/ascend/examples/pytest_ut --ignore=third_party/ascend/examples/pytest_ut/test_index_select.py \ | |
| --ignore=third_party/ascend/examples/pytest_ut/test_linearize_permute.py \ | |
| --ignore=third_party/ascend/examples/pytest_ut/test_logical_and.py \ | |
| --ignore=third_party/ascend/examples/pytest_ut/test_logical_or.py \ | |
| --ignore=third_party/ascend/examples/pytest_ut/test_triton_unified_attention.py |