Skip to content

[CI] Reference main branch workflows for triton_v3.3.x branch #341

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
youngyoung01 wants to merge 2 commits into
base: triton_v3.3.x
Choose a base branch
from
Open

[CI] Reference main branch workflows for triton_v3.3.x branch #341

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a780ea0
[CI] Reference main branch workflows for triton_v3.3.x branch
youngyoung01 Feb 3, 2026
6901ab5
Merge branch 'triton_v3.3.x' into workflow_triton_v3.3.x
youngyoung01 Feb 3, 2026
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
94 changes: 1 addition & 93 deletions .github/workflows/aipu-build-and-test.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,98 +6,6 @@ on:
pull_request:
branches: [ "triton_v3.3.x" ]

concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
cancel-in-progress: true

jobs:
aipu-build-and-test:
runs-on: aipu
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 AIPU
if: steps.check_files.outputs.only_docs_changed != 'true'
shell: bash
run: |
set -x
pip uninstall -y triton
source ~/env_setup.sh
export FLAGTREE_BACKEND=aipu
cd python
MAX_JOBS=32 python3.10 -m pip install . --no-build-isolation

- name: FlagTree Test on AIPU
if: steps.check_files.outputs.only_docs_changed != 'true'
shell: bash
run: |
set -x
source ~/env_setup.sh
python3.10 third_party/aipu/python/test/test_01_vector_add.py
python3.10 third_party/aipu/python/test/test_01_vector_add_shared_memory.py
python3.10 third_party/aipu/python/test/test_01_vector_add_dma.py
python3.10 third_party/aipu/python/test/test_02_fused_softmax.py
python3.10 third_party/aipu/python/test/test_02_fused_softmax_shared_memory.py
python3.10 third_party/aipu/python/test/test_02_fused_softmax_dma.py
python3.10 third_party/aipu/python/test/test_libdevice_fmod.py
python3.10 third_party/aipu/python/test/test_libdevice_fmod_shared_memory.py
python3.10 third_party/aipu/python/test/test_libdevice_fmod_dma.py
python3.10 third_party/aipu/python/test/test_libdevice_pow.py
python3.10 third_party/aipu/python/test/test_libdevice_pow_shared_memory.py
python3.10 third_party/aipu/python/test/test_libdevice_pow_dma.py
python3.10 third_party/aipu/python/test/test_codegen_dma.py
python3.10 third_party/aipu/python/test/test_01_vector_add_structured_1d.py
python3.10 third_party/aipu/python/test/test_01_vector_add_structured_1d_shared_memory.py
python3.10 third_party/aipu/python/test/test_01_vector_add_structured_1d_dma.py
python3.10 third_party/aipu/python/test/min.py
python3.10 third_party/aipu/python/test/min_shared_memory.py
#python3.10 third_party/aipu/python/test/min_dma.py
python3.10 third_party/aipu/python/test/var_mean.py
python3.10 third_party/aipu/python/test/var_mean_shared_memory.py
#python3.10 third_party/aipu/python/test/var_mean_dma.py
uses: flagos-ai/flagtree/.github/workflows/aipu-build-and-test.yml@workflow_unifiy
69 changes: 1 addition & 68 deletions .github/workflows/ascend-build-and-test.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,73 +6,6 @@ on:
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: ascend
steps:
- name: Checkout code (attempt 1)
id: checkout1
uses: actions/checkout@v5
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@v5
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@v5

- name: Verify checkout success
if: success()
run: echo "Checkout completed successfully"

- name: FlagTree Build on Ascend
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
shell: bash
run: |
set -x
source /usr/local/Ascend/ascend-toolkit/set_env.sh
python3 third_party/tests/ascend/vector-add.py
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/05-matrix-multiplication.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/08-demo-libentry.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/12-hstu_attention.py
python3 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized-flagtree.py
python3 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized.py
python3 third_party/ascend/examples/tutorials/14-accuracy-comparison.py
uses: flagos-ai/flagtree/.github/workflows/ascend-build-and-test.yml@workflow_unifiy
53 changes: 1 addition & 52 deletions .github/workflows/cambricon-build-and-test.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,57 +6,6 @@ on:
pull_request:
branches: [ "main" ]

concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
cancel-in-progress: true

jobs:
cambricon-build-and-test:
runs-on: cambricon
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@v5
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@v5
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@v5

- name: Verify checkout success
if: success()
run: echo "Checkout completed successfully"

- name: FlagTree Build on Cambricon
shell: bash
run: |
set -x
pip uninstall -y triton
export FLAGTREE_BACKEND=cambricon
cd python
MAX_JOBS=8 python3 -m pip install . --no-build-isolation
uses: flagos-ai/flagtree/.github/workflows/cambricon-build-and-test.yml@workflow_unifiy
72 changes: 1 addition & 71 deletions .github/workflows/enflame-build-and-test.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,76 +6,6 @@ on:
pull_request:
branches: [ "triton_v3.3.x" ]

concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
cancel-in-progress: true

jobs:
enflame-build-and-test:
runs-on: enflame
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 Enflame
if: steps.check_files.outputs.only_docs_changed != 'true'
shell: bash
run: |
set -x
pip uninstall -y triton
pip uninstall -y triton_gcu
export FLAGTREE_BACKEND=enflame
cd python
MAX_JOBS=32 python3 -m pip install . --no-build-isolation

- name: FlagTree Test on Enflame
if: steps.check_files.outputs.only_docs_changed != 'true'
shell: bash
run: |
set -x
python3 -m pytest -s third_party/enflame/python/test/unit
uses: flagos-ai/flagtree/.github/workflows/enflame-build-and-test.yml@workflow_unifiy
60 changes: 1 addition & 59 deletions .github/workflows/hcu-build-and-test.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,64 +6,6 @@ on:
pull_request:
branches: [ "main" ]

concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
cancel-in-progress: true

jobs:
hcu-build-and-test:
runs-on: hcu
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@v5
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@v5
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@v5

- name: Verify checkout success
if: success()
run: echo "Checkout completed successfully"

- name: FlagTree Build on Hcu
shell: bash
run: |
set -x
pip uninstall -y triton
export FLAGTREE_BACKEND=hcu
cd python
MAX_JOBS=32 python3 -m pip install . --no-build-isolation

- name: FlagTree Test on Hcu
shell: bash
run: |
set -x
cd third_party/hcu/python/test/unit
sh flagtree_test.sh
uses: flagos-ai/flagtree/.github/workflows/hcu-build-and-test.yml@workflow_unifiy
Loading
Loading