Skip to content

Commit f237c73

Browse files
committed
[CI] Update nv workflow
1 parent 16da717 commit f237c73

1 file changed

Lines changed: 13 additions & 6 deletions

File tree

.github/workflows/nv-build-and-test.yml

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ concurrency:
1414

1515
jobs:
1616
nv-build-and-test:
17-
runs-on: nv-jiuding
17+
runs-on: flagtree-nvidia
1818
if: ${{ github.repository == 'FlagTree/flagtree' || github.repository == 'flagos-ai/flagtree' }}
1919
steps:
2020
- name: Setup environment
@@ -26,6 +26,8 @@ jobs:
2626
- name: Checkout code (attempt 1)
2727
id: checkout1
2828
uses: actions/checkout@v6
29+
with:
30+
fetch-depth: 0
2931
continue-on-error: true
3032

3133
- name: Sleep before checkout2
@@ -38,6 +40,8 @@ jobs:
3840
id: checkout2
3941
if: steps.checkout1.outcome == 'failure'
4042
uses: actions/checkout@v6
43+
with:
44+
fetch-depth: 0
4145
continue-on-error: true
4246

4347
- name: Sleep before final checkout
@@ -49,6 +53,8 @@ jobs:
4953
- name: Checkout code (final attempt)
5054
if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
5155
uses: actions/checkout@v6
56+
with:
57+
fetch-depth: 0
5258

5359
- name: Verify checkout success
5460
if: success()
@@ -78,7 +84,7 @@ jobs:
7884
pip uninstall -y triton
7985
source ~/env.sh
8086
cd python
81-
MAX_JOBS=32 python3.11 -m pip install . --no-build-isolation
87+
MAX_JOBS=32 python3 -m pip install . --no-build-isolation
8288
8389
- name: FlagTree Build on NVidia (triton_v3.2.x branch)
8490
if: ${{ steps.check_files.outputs.only_docs_changed != 'true' && env.TARGET_BRANCH == 'triton_v3.2.x' }}
@@ -88,7 +94,7 @@ jobs:
8894
pip uninstall -y triton
8995
source ~/env-3.2.sh
9096
cd python
91-
MAX_JOBS=32 python3.11 -m pip install . --no-build-isolation
97+
MAX_JOBS=32 python3 -m pip install . --no-build-isolation
9298
9399
- name: FlagTree Build on NVidia (triton_v3.3.x branch)
94100
if: ${{ steps.check_files.outputs.only_docs_changed != 'true' && env.TARGET_BRANCH == 'triton_v3.3.x' }}
@@ -98,14 +104,15 @@ jobs:
98104
pip uninstall -y triton
99105
source ~/env-3.3.sh
100106
cd python
101-
MAX_JOBS=32 python3.11 -m pip install . --no-build-isolation
107+
MAX_JOBS=32 python3 -m pip install . --no-build-isolation
102108
103109
- name: FlagTree Test on NVidia
104110
if: steps.check_files.outputs.only_docs_changed != 'true'
105111
shell: bash
106112
run: |
107113
set -x
108-
python3.11 -m pytest -s python/test/unit
114+
python3 -m pytest -s python/test/unit \
115+
--ignore=python/test/unit/language/test_line_info.py
109116
if [ -d "python/test/operators" ]; then
110-
python3.11 -m pytest -s python/test/operators
117+
python3 -m pytest -s python/test/operators
111118
fi

0 commit comments

Comments
 (0)