Skip to content

Commit 92531e0

Browse files
authored
Merge branch 'main' into ko3n1g/docs/pipeline-trigger-pip-install-deps
2 parents 9257687 + f8becec commit 92531e0

File tree

213 files changed

+122621
-81666
lines changed

Some content is hidden

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

213 files changed

+122621
-81666
lines changed

.github/actions/action.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ runs:
5959
run: echo "node_name=$NODE_NAME" | tee -a "$GITHUB_OUTPUT"
6060

6161
- name: Checkout repository
62-
uses: actions/checkout@v2
62+
uses: actions/checkout@v6
6363

6464
- name: Change ownership of /home/runner/
6565
shell: bash

.github/copy-pr-bot.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
enabled: true
22
auto_sync_draft: false
33
auto_sync_ready: true
4-
trustees_override: ["AAnoosheh", "ArEsKay3", "Autumn1998", "BestJuly", "BoxiangW", "CarlosGomes98", "ChenhanYu", "FDecaYed", "HaochenYuan", "ISEEKYAN", "JRD971000", "Phlip79", "QiZhangNV", "RPrenger", "ShriyaRishab", "Victarry", "Wohox", "ZhiyuLi-Nvidia", "ahmadki", "aklife97", "ananthsub", "asolergi-nv", "buptzyb", "chtruong814", "cjld", "cspades", "cuichenx", "deepakn94", "dimapihtar", "dingqingy-nv", "duncanriach", "erhoo82", "ericharper", "fanshiqing", "faradawn", "frsun-nvda", "gautham-kollu", "gdengk", "guyueh1", "hxbai", "ilml", "jalbericiola", "janEbert", "jaredcasper", "jenchen13", "jiemingz", "jingqiny-99", "jkamalu", "jon-barker", "jstjohn", "kanz-nv", "kevalmorabia97", "ko3n1g", "kunlunl", "kvareddy", "kwyss-nvidia", "layalir", "lhb8125", "lmcafee-nvidia", "maanug-nv", "mathemakitten", "matthieule", "mchrzanowski", "mehraakash", "mkhona-nvidia", "nanz-nv", "parthmannan", "prajwal1210", "pthombre", "rogerwaleffe", "sajadn", "sanandaraj5597", "sancha", "santhnm2", "sbak5", "shanmugamr1992", "sharathts", "shengf-nv", "shifangx", "shjwudp", "sidsingh-nvidia", "skyw", "sudhakarsingh27", "tdene", "theothermike", "thomasdhc", "tomlifu", "trintamaki", "tylerpoon", "wdykas", "xiaoyao0115", "xuwchen", "yanring", "yaox12", "yaoyu-33", "yashaswikarnati", "yeyu-nvidia", "yobibyte", "youngeunkwon0405", "yueshen2016", "yuzhongw-nvidia", "zhongbozhu"]
4+
trustees_override: ["AAnoosheh", "ArEsKay3", "Autumn1998", "BestJuly", "BoxiangW", "CarlosGomes98", "ChenhanYu", "FDecaYed", "HaochenYuan", "ISEEKYAN", "JRD971000", "Phlip79", "QiZhangNV", "RPrenger", "ShriyaRishab", "Victarry", "Wohox", "ZhiyuLi-Nvidia", "ahmadki", "aklife97", "ananthsub", "asolergi-nv", "buptzyb", "chtruong814", "cjld", "cspades", "cuichenx", "deepakn94", "dimapihtar", "dingqingy-nv", "duncanriach", "erhoo82", "ericharper", "fanshiqing", "faradawn", "frsun-nvda", "gautham-kollu", "gdengk", "guyueh1", "huvunvidia", "hxbai", "ilml", "jalbericiola", "janEbert", "jaredcasper", "jenchen13", "jiemingz", "jingqiny-99", "jkamalu", "jon-barker", "jstjohn", "kanz-nv", "kevalmorabia97", "ko3n1g", "ksivaman", "kunlunl", "kvareddy", "kwyss-nvidia", "layalir", "lhb8125", "lmcafee-nvidia", "maanug-nv", "mathemakitten", "matthieule", "mchrzanowski", "mehraakash", "mkhona-nvidia", "nanz-nv", "parthmannan", "prajwal1210", "pthombre", "rhewett-nv", "rogerwaleffe", "sajadn", "sanandaraj5597", "sancha", "santhnm2", "sbak5", "shanmugamr1992", "sharathts", "shengf-nv", "shifangx", "shjwudp", "sidsingh-nvidia", "skyw", "sudhakarsingh27", "tdene", "theothermike", "thomasdhc", "tomlifu", "trintamaki", "tylerpoon", "wdykas", "wplf", "xiaoyao0115", "xuwchen", "yanring", "yaox12", "yaoyu-33", "yashaswikarnati", "yeyu-nvidia", "yobibyte", "youngeunkwon0405", "yueshen2016", "yuzhongw-nvidia", "zhongbozhu"]

.github/oncall_schedule.json

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,4 @@
11
[
2-
{
3-
"user": "BoxiangW",
4-
"date": "2026-03-04"
5-
},
62
{
73
"user": "maanug-nv",
84
"date": "2026-03-11"
@@ -12,11 +8,11 @@
128
"date": "2026-03-18"
139
},
1410
{
15-
"user": "gautham-kollu",
11+
"user": "janEbert",
1612
"date": "2026-03-25"
1713
},
1814
{
19-
"user": "janEbert",
15+
"user": "gautham-kollu",
2016
"date": "2026-04-01"
2117
},
2218
{
@@ -46,5 +42,9 @@
4642
{
4743
"user": "gautham-kollu",
4844
"date": "2026-05-20"
45+
},
46+
{
47+
"user": "ilml",
48+
"date": "2026-05-27"
4949
}
5050
]

.github/workflows/_build_test_publish_wheel.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ jobs:
4343
PUBLISH_DRYRUN: ${{ inputs.dry-run }}
4444
steps:
4545
- name: Checkout repository
46-
uses: actions/checkout@v4
46+
uses: actions/checkout@v6
4747
with:
4848
ref: ${{ inputs.ref }}
4949

@@ -136,7 +136,7 @@ jobs:
136136
test "${{ steps.build-wheel.outputs.expected-release-number }}" == "$RELEASE_NUMBER"
137137
138138
- name: Upload wheels
139-
uses: actions/upload-artifact@v4
139+
uses: actions/upload-artifact@v6
140140
with:
141141
name: wheels-${{ matrix.PACKAGE }}-${{ matrix.PLATFORM }}-${{ inputs.dry-run && 'dry-run' || 'release' }}
142142
path: dist/
@@ -159,7 +159,7 @@ jobs:
159159
PACKAGE: ${{ matrix.PACKAGE }}
160160
steps:
161161
- name: Download wheels
162-
uses: actions/download-artifact@v4
162+
uses: actions/download-artifact@v7
163163
with:
164164
name: wheels-${{ matrix.PACKAGE }}-${{ matrix.PLATFORM }}-${{ inputs.dry-run && 'dry-run' || 'release' }}
165165
path: dist/

.github/workflows/_release_library.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ jobs:
110110
IS_DRY_RUN: ${{ inputs.dry-run }}
111111
steps:
112112
- name: Checkout repository
113-
uses: actions/checkout@v4
113+
uses: actions/checkout@v6
114114
with:
115115
path: ${{ github.run_id }}
116116
token: ${{ secrets.PAT }}
@@ -221,7 +221,7 @@ jobs:
221221
PR_NUMBER=$(echo $PR_URL | grep -o '[0-9]*$')
222222
223223
- name: Wait for status checks on tmp branch
224-
uses: actions/github-script@v7
224+
uses: actions/github-script@v8
225225
id: wait-status
226226
with:
227227
github-token: ${{ secrets.PAT }}
@@ -364,7 +364,7 @@ jobs:
364364
TAG_PREFIX: core_
365365
steps:
366366
- name: Checkout repository
367-
uses: actions/checkout@v4
367+
uses: actions/checkout@v6
368368
with:
369369
path: ${{ github.run_id }}
370370
ref: ${{ inputs.release-ref }}
@@ -495,7 +495,7 @@ jobs:
495495
VERSION: ${{ needs.build-test-publish-wheels.outputs.version }}
496496
steps:
497497
- name: Checkout
498-
uses: actions/checkout@v4
498+
uses: actions/checkout@v6
499499
with:
500500
repository: NVIDIA-NeMo/FW-CI-templates
501501
ref: v0.17.0

.github/workflows/_update_dependencies.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ jobs:
3333
TARGET_BRANCH: ${{ inputs.target-branch }}
3434
steps:
3535
- name: Checkout repo
36-
uses: actions/checkout@v4
36+
uses: actions/checkout@v6
3737
with:
3838
ref: ${{ env.TARGET_BRANCH }}
3939

@@ -60,7 +60,7 @@ jobs:
6060
fi
6161
6262
- name: Checkout repo
63-
uses: actions/checkout@v4
63+
uses: actions/checkout@v6
6464
with:
6565
ref: ${{ env.SOURCE_BRANCH }}
6666

@@ -77,7 +77,7 @@ jobs:
7777
bash -c 'uv lock --upgrade'
7878
7979
- name: Upload lock file
80-
uses: actions/upload-artifact@v4
80+
uses: actions/upload-artifact@v6
8181
with:
8282
name: lock-file-${{ env.SOURCE_BRANCH }}
8383
path: uv.lock
@@ -90,7 +90,7 @@ jobs:
9090
TARGET_BRANCH: ${{ inputs.target-branch }}
9191
steps:
9292
- name: Checkout code
93-
uses: actions/checkout@v4
93+
uses: actions/checkout@v6
9494
with:
9595
token: ${{ secrets.PAT }}
9696
ref: ${{ env.TARGET_BRANCH }}
@@ -103,7 +103,7 @@ jobs:
103103
fi
104104
105105
- name: Download lock file
106-
uses: actions/download-artifact@v4
106+
uses: actions/download-artifact@v7
107107
with:
108108
name: lock-file-${{ env.SOURCE_BRANCH }}
109109

.github/workflows/auto-reminder-bot.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,10 @@ jobs:
1414
if: github.repository == 'NVIDIA/Megatron-LM'
1515
steps:
1616
- name: Check out repository code
17-
uses: actions/checkout@v4
17+
uses: actions/checkout@v6
1818

1919
- name: Set up Python
20-
uses: actions/setup-python@v5
20+
uses: actions/setup-python@v6
2121
with:
2222
python-version: "3.10"
2323

.github/workflows/auto-swap-labels.yml

Lines changed: 41 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -2,42 +2,74 @@
22

33
name: Auto Swap Labels
44
on:
5-
pull_request_review:
6-
types: [submitted]
75
pull_request_target:
86
types: [ready_for_review, synchronize]
97
branches:
108
- main
9+
workflow_run:
10+
workflows: ["Review Trigger"]
11+
types: [completed]
1112

1213
permissions:
1314
pull-requests: write
1415
contents: read
16+
actions: read
1517

1618
jobs:
1719
check-approval:
1820
runs-on: ubuntu-latest
1921
if: >-
20-
github.repository == 'NVIDIA/Megatron-LM' &&
21-
github.event.pull_request.base.ref == 'main' &&
22-
(
23-
(github.event_name == 'pull_request_review' && github.event.review.state == 'approved') ||
24-
(github.event_name == 'pull_request_target' && !github.event.pull_request.draft)
22+
github.repository == 'NVIDIA/Megatron-LM' && (
23+
(github.event_name == 'pull_request_target' &&
24+
github.event.pull_request.base.ref == 'main' &&
25+
!github.event.pull_request.draft) ||
26+
(github.event_name == 'workflow_run' &&
27+
github.event.workflow_run.conclusion == 'success')
2528
)
29+
2630
steps:
31+
- name: Get PR number from workflow_run
32+
id: get-pr
33+
if: github.event_name == 'workflow_run'
34+
continue-on-error: true
35+
uses: actions/download-artifact@v4
36+
with:
37+
name: pr-number
38+
path: pr-number
39+
github-token: ${{ github.token }}
40+
run-id: ${{ github.event.workflow_run.id }}
41+
42+
- name: Set PR number
43+
id: pr
44+
run: |
45+
if [ "${{ github.event_name }}" = "workflow_run" ]; then
46+
if [ "${{ steps.get-pr.outcome }}" != "success" ]; then
47+
echo "No approval artifact found — review was not an approval. Skipping."
48+
exit 0
49+
fi
50+
echo "number=$(cat pr-number/number)" >> $GITHUB_OUTPUT
51+
else
52+
echo "number=${{ github.event.pull_request.number }}" >> $GITHUB_OUTPUT
53+
fi
54+
2755
- name: Check out repository code
56+
if: steps.pr.outputs.number
2857
uses: actions/checkout@v4
2958

3059
- name: Set up Python
31-
uses: actions/setup-python@v5
60+
if: steps.pr.outputs.number
61+
uses: actions/setup-python@v6
3262
with:
3363
python-version: "3.10"
3464

3565
- name: Install dependencies
66+
if: steps.pr.outputs.number
3667
run: |
3768
pip install --no-cache-dir PyGithub slack-sdk
3869
3970
- name: Run Auto Swap Labels
71+
if: steps.pr.outputs.number
4072
run: |
4173
export GH_TOKEN=${{ secrets.PAT }}
42-
export PR_NUMBER=${{ github.event.pull_request.number }}
74+
export PR_NUMBER=${{ steps.pr.outputs.number }}
4375
python tests/test_utils/python_scripts/swap_pr_labels.py

.github/workflows/auto-update-copy-pr-bot.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ jobs:
1111
if: github.repository == 'NVIDIA/Megatron-LM'
1212
steps:
1313
- name: Checkout code
14-
uses: actions/checkout@v3
14+
uses: actions/checkout@v6
1515
with:
1616
token: ${{ secrets.PAT }}
1717
ref: main

.github/workflows/build-test-publish-wheel.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ jobs:
5858
runs-on: ubuntu-latest
5959
steps:
6060
- name: Checkout repository
61-
uses: actions/checkout@v4
61+
uses: actions/checkout@v6
6262

6363
- name: Result
6464
env:

0 commit comments

Comments
 (0)