Skip to content

Commit 018a6e0

Browse files
sgjzfzzfsgjzfzzf
authored
Merge branch 'triton_v3.2.x' into triton_v3.2.x_hint_manager
2 parents 5f7a336 + b3091e9 commit 018a6e0

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

59 files changed

+3415
-347
lines changed

‎.github/workflows/aipu-build-and-test.yml‎

Lines changed: 4 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -21,36 +21,10 @@ jobs:
2121
source ~/env.sh
2222
env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true
2323
24-
- name: Checkout code (attempt 1)
25-
id: checkout1
26-
uses: actions/checkout@v5
27-
continue-on-error: true
28-
29-
- name: Sleep before checkout2
30-
if: steps.checkout1.outcome == 'failure'
31-
run: |
32-
echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
33-
sleep 120
34-
35-
- name: Checkout code (attempt 2)
36-
id: checkout2
37-
if: steps.checkout1.outcome == 'failure'
38-
uses: actions/checkout@v5
39-
continue-on-error: true
40-
41-
- name: Sleep before final checkout
42-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
43-
run: |
44-
echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
45-
sleep 180
46-
47-
- name: Checkout code (final attempt)
48-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
49-
uses: actions/checkout@v5
50-
51-
- name: Verify checkout success
52-
if: success()
53-
run: echo "Checkout completed successfully"
24+
- name: Smart Checkout
25+
uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
26+
with:
27+
checkout_version: 'v6'
5428

5529
- name: FlagTree Build on AIPU
5630
shell: bash

‎.github/workflows/ascend-build-and-test.yml‎

Lines changed: 13 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -21,42 +21,10 @@ jobs:
2121
source ~/env.sh
2222
env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true
2323
24-
- name: Checkout code (attempt 1)
25-
id: checkout1
26-
uses: actions/checkout@v6
24+
- name: Smart Checkout
25+
uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
2726
with:
28-
fetch-depth: 0
29-
continue-on-error: true
30-
31-
- name: Sleep before checkout2
32-
if: steps.checkout1.outcome == 'failure'
33-
run: |
34-
echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
35-
sleep 120
36-
37-
- name: Checkout code (attempt 2)
38-
id: checkout2
39-
if: steps.checkout1.outcome == 'failure'
40-
uses: actions/checkout@v6
41-
with:
42-
fetch-depth: 0
43-
continue-on-error: true
44-
45-
- name: Sleep before final checkout
46-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
47-
run: |
48-
echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
49-
sleep 180
50-
51-
- name: Checkout code (final attempt)
52-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
53-
uses: actions/checkout@v6
54-
with:
55-
fetch-depth: 0
56-
57-
- name: Verify checkout success
58-
if: success()
59-
run: echo "Checkout completed successfully"
27+
checkout_version: 'v6'
6028

6129
- name: Check if only docs files changed
6230
id: check_files
@@ -104,8 +72,16 @@ jobs:
10472
--ignore=test_assume.py \
10573
--ignore=test_index_select.py
10674
popd
75+
# flagtree tle test
76+
pushd python/test/tle
77+
python3 test_vec_add.py
78+
python3 test_vec_add_2d.py
79+
python3 test_vec_add_mix.py
80+
python3 test_vec_mathOps.py
81+
python3 test_tle_with_hints.py
82+
popd
10783
108-
- name: FlagTree Editable Build And Teston Ascend
84+
- name: FlagTree Editable Build And Test on Ascend
10985
if: steps.check_files.outputs.only_docs_changed != 'true'
11086
shell: bash
11187
run: |
@@ -118,3 +94,4 @@ jobs:
11894
popd
11995
source /usr/local/Ascend/ascend-toolkit/set_env.sh
12096
python3 -c "import triton"
97+
python3 -m pip uninstall flagtree -y

‎.github/workflows/cambricon-build-and-test.yml‎

Lines changed: 4 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -21,36 +21,10 @@ jobs:
2121
source ~/env.sh
2222
env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true
2323
24-
- name: Checkout code (attempt 1)
25-
id: checkout1
26-
uses: actions/checkout@v5
27-
continue-on-error: true
28-
29-
- name: Sleep before checkout2
30-
if: steps.checkout1.outcome == 'failure'
31-
run: |
32-
echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
33-
sleep 120
34-
35-
- name: Checkout code (attempt 2)
36-
id: checkout2
37-
if: steps.checkout1.outcome == 'failure'
38-
uses: actions/checkout@v5
39-
continue-on-error: true
40-
41-
- name: Sleep before final checkout
42-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
43-
run: |
44-
echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
45-
sleep 180
46-
47-
- name: Checkout code (final attempt)
48-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
49-
uses: actions/checkout@v5
50-
51-
- name: Verify checkout success
52-
if: success()
53-
run: echo "Checkout completed successfully"
24+
- name: Smart Checkout
25+
uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
26+
with:
27+
checkout_version: 'v6'
5428

5529
- name: FlagTree Build on Cambricon
5630
shell: bash

‎.github/workflows/code_scan.yml‎

Lines changed: 4 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -21,36 +21,10 @@ jobs:
2121
source ~/env.sh
2222
env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true
2323
24-
- name: Checkout code (attempt 1)
25-
id: checkout1
26-
uses: actions/checkout@v5
27-
continue-on-error: true
28-
29-
- name: Sleep before checkout2
30-
if: steps.checkout1.outcome == 'failure'
31-
run: |
32-
echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
33-
sleep 120
34-
35-
- name: Checkout code (attempt 2)
36-
id: checkout2
37-
if: steps.checkout1.outcome == 'failure'
38-
uses: actions/checkout@v5
39-
continue-on-error: true
40-
41-
- name: Sleep before final checkout
42-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
43-
run: |
44-
echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
45-
sleep 180
46-
47-
- name: Checkout code (final attempt)
48-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
49-
uses: actions/checkout@v5
50-
51-
- name: Verify checkout success
52-
if: success()
53-
run: echo "Checkout completed successfully"
24+
- name: Smart Checkout
25+
uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
26+
with:
27+
checkout_version: 'v6'
5428

5529
- name: Scan flagtree repo
5630
shell: bash

‎.github/workflows/hcu-build-and-test.yml‎

Lines changed: 4 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -21,36 +21,10 @@ jobs:
2121
source ~/env.sh
2222
env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true
2323
24-
- name: Checkout code (attempt 1)
25-
id: checkout1
26-
uses: actions/checkout@v5
27-
continue-on-error: true
28-
29-
- name: Sleep before checkout2
30-
if: steps.checkout1.outcome == 'failure'
31-
run: |
32-
echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
33-
sleep 120
34-
35-
- name: Checkout code (attempt 2)
36-
id: checkout2
37-
if: steps.checkout1.outcome == 'failure'
38-
uses: actions/checkout@v5
39-
continue-on-error: true
40-
41-
- name: Sleep before final checkout
42-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
43-
run: |
44-
echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
45-
sleep 180
46-
47-
- name: Checkout code (final attempt)
48-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
49-
uses: actions/checkout@v5
50-
51-
- name: Verify checkout success
52-
if: success()
53-
run: echo "Checkout completed successfully"
24+
- name: Smart Checkout
25+
uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
26+
with:
27+
checkout_version: 'v6'
5428

5529
- name: FlagTree Build on Hcu
5630
shell: bash

‎.github/workflows/iluvatar-build-and-test.yml‎

Lines changed: 4 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -15,36 +15,10 @@ jobs:
1515
runs-on: iluvatar
1616
if: ${{ github.repository == 'FlagTree/flagtree' || github.repository == 'flagos-ai/flagtree' }}
1717
steps:
18-
- name: Checkout code (attempt 1)
19-
id: checkout1
20-
uses: actions/checkout@v5
21-
continue-on-error: true
22-
23-
- name: Sleep before checkout2
24-
if: steps.checkout1.outcome == 'failure'
25-
run: |
26-
echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
27-
sleep 120
28-
29-
- name: Checkout code (attempt 2)
30-
id: checkout2
31-
if: steps.checkout1.outcome == 'failure'
32-
uses: actions/checkout@v5
33-
continue-on-error: true
34-
35-
- name: Sleep before final checkout
36-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
37-
run: |
38-
echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
39-
sleep 180
40-
41-
- name: Checkout code (final attempt)
42-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
43-
uses: actions/checkout@v5
44-
45-
- name: Verify checkout success
46-
if: success()
47-
run: echo "Checkout completed successfully"
18+
- name: Smart Checkout
19+
uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
20+
with:
21+
checkout_version: 'v6'
4822

4923
- name: FlagTree Build on Iluvatar
5024
shell: bash

‎.github/workflows/metax-build-and-test.yml‎

Lines changed: 4 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -15,36 +15,10 @@ jobs:
1515
runs-on: metax
1616
if: ${{ github.repository == 'FlagTree/flagtree' || github.repository == 'flagos-ai/flagtree' }}
1717
steps:
18-
- name: Checkout code (attempt 1)
19-
id: checkout1
20-
uses: actions/checkout@v5
21-
continue-on-error: true
22-
23-
- name: Sleep before checkout2
24-
if: steps.checkout1.outcome == 'failure'
25-
run: |
26-
echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
27-
sleep 120
28-
29-
- name: Checkout code (attempt 2)
30-
id: checkout2
31-
if: steps.checkout1.outcome == 'failure'
32-
uses: actions/checkout@v5
33-
continue-on-error: true
34-
35-
- name: Sleep before final checkout
36-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
37-
run: |
38-
echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
39-
sleep 180
40-
41-
- name: Checkout code (final attempt)
42-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
43-
uses: actions/checkout@v5
44-
45-
- name: Verify checkout success
46-
if: success()
47-
run: echo "Checkout completed successfully"
18+
- name: Smart Checkout
19+
uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
20+
with:
21+
checkout_version: 'v6'
4822

4923
- name: FlagTree Build on Metax
5024
shell: bash

‎.github/workflows/mthreads-build-and-test.yml‎

Lines changed: 4 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -21,36 +21,10 @@ jobs:
2121
source ~/env.sh
2222
env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true
2323
24-
- name: Checkout code (attempt 1)
25-
id: checkout1
26-
uses: actions/checkout@v5
27-
continue-on-error: true
28-
29-
- name: Sleep before checkout2
30-
if: steps.checkout1.outcome == 'failure'
31-
run: |
32-
echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
33-
sleep 120
34-
35-
- name: Checkout code (attempt 2)
36-
id: checkout2
37-
if: steps.checkout1.outcome == 'failure'
38-
uses: actions/checkout@v5
39-
continue-on-error: true
40-
41-
- name: Sleep before final checkout
42-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
43-
run: |
44-
echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
45-
sleep 180
46-
47-
- name: Checkout code (final attempt)
48-
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
49-
uses: actions/checkout@v5
50-
51-
- name: Verify checkout success
52-
if: success()
53-
run: echo "Checkout completed successfully"
24+
- name: Smart Checkout
25+
uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
26+
with:
27+
checkout_version: 'v6'
5428

5529
- name: FlagTree Build on Mthreads
5630
shell: bash

0 commit comments

Comments
 (0)