Skip to content

Commit 12cf77b

Browse files
lhb8125lhb8125
committed
Merge branch 'main' into hongbinl/offload_activation_cuda_graph
2 parents 7ca3618 + 59f6f38 commit 12cf77b

File tree

724 files changed

+30410
-6411
lines changed

Some content is hidden

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

724 files changed

+30410
-6411
lines changed

‎.github/actions/build-pytorch-wheel/Dockerfile‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# See LICENSE for license information.
44

‎.github/actions/build-pytorch-wheel/action.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# See LICENSE for license information.
44

‎.github/actions/build-pytorch-wheel/build.sh‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#!/bin/bash
22

3-
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3+
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
44
#
55
# See LICENSE for license information.
66

‎.github/scripts/check_for_ngc_images.sh‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#!/bin/bash
22

3-
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3+
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
44
#
55
# See LICENSE for license information.
66

‎.github/workflows/attach-wheels-to-release.yml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# See LICENSE for license information.
44

@@ -96,7 +96,7 @@ jobs:
9696
9797
- name: Get Release with tag
9898
id: get_current_release
99-
uses: joutvhu/get-release@v1
99+
uses: joutvhu/get-release@9a8271732adc3299a22f8ad09b0a67eb3aa836ac
100100
if: ${{ github.event_name == 'workflow_dispatch' }}
101101
with:
102102
tag_name: ${{ inputs.release-version }}

‎.github/workflows/blossom-ci.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# See LICENSE for license information.
44

‎.github/workflows/build.yml‎

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# See LICENSE for license information.
44

@@ -24,11 +24,14 @@ jobs:
2424
uses: actions/checkout@v3
2525
with:
2626
submodules: recursive
27+
- name: ccache
28+
uses: mozilla-actions/sccache-action@7d986dd989559c6ecdb630a3fd2557667be217ad
2729
- name: 'Build'
28-
run: pip install --no-build-isolation . -v
30+
run: NVTE_USE_CCACHE=1 NVTE_CCACHE_BIN=sccache pip install --no-build-isolation . -v
2931
env:
3032
NVTE_FRAMEWORK: none
3133
MAX_JOBS: 1
34+
SCCACHE_GHA_ENABLED: "true"
3235
- name: 'Sanity check'
3336
run: python3 -c "import transformer_engine"
3437
working-directory: /
@@ -89,16 +92,20 @@ jobs:
8992
options: --user root
9093
steps:
9194
- name: 'Dependencies'
92-
run: pip install pybind11[global]
95+
run: pip install cmake==3.21.0 pybind11[global]
9396
- name: 'Checkout'
9497
uses: actions/checkout@v3
9598
with:
9699
submodules: recursive
100+
- name: ccache
101+
uses: mozilla-actions/sccache-action@7d986dd989559c6ecdb630a3fd2557667be217ad
97102
- name: 'Build'
98-
run: pip install --no-build-isolation . -v
103+
run: |
104+
NVTE_CCACHE_BIN=sccache NVTE_USE_CCACHE=1 pip install --no-build-isolation . -v
99105
env:
100106
NVTE_FRAMEWORK: jax
101107
MAX_JOBS: 1
108+
SCCACHE_GHA_ENABLED: "true"
102109
- name: 'Sanity check'
103110
run: python3 tests/jax/test_sanity_import.py
104111
all:
@@ -137,10 +144,9 @@ jobs:
137144
- name: 'Dependencies'
138145
run: |
139146
docker exec builder bash -c '\
140-
pip install pybind11[global] einops onnxscript && \
147+
pip install cmake==3.21.0 pybind11[global] einops onnxscript && \
141148
pip install torch --no-cache-dir --index-url https://download.pytorch.org/whl/cu130
142149
'
143-
144150
- name: 'Build'
145151
run: docker exec builder bash -c 'pip install --no-cache-dir --no-build-isolation . -v --no-deps'
146152
env:

‎.github/workflows/deploy_nightly_docs.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# See LICENSE for license information.
44

‎.github/workflows/docs.yml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# See LICENSE for license information.
44

@@ -17,7 +17,7 @@ jobs:
1717
uses: actions/checkout@v3
1818
- name: 'Install dependencies'
1919
run: |
20-
pip install sphinx==8.1.3 sphinx_rtd_theme==3.0.1 nbsphinx==0.9.5 IPython ipython_genutils==0.2.0 ipywidgets==8.0.2 astroid==3.3.2
20+
pip install sphinx==8.1.3 sphinx_rtd_theme==3.0.1 nbsphinx==0.9.5 IPython ipython_genutils==0.2.0 ipywidgets==8.0.2 astroid==3.3.2 sphinx-tabs==3.4.7
2121
pip install breathe==4.35.0 sphinx-autoapi==3.3.2
2222
sudo apt-get install -y pandoc graphviz doxygen
2323
export GIT_SHA=$(git show-ref --hash HEAD)

‎.github/workflows/license.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# See LICENSE for license information.
44

0 commit comments

Comments
 (0)