Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
49 commits
Select commit Hold shift + click to select a range
43206d9
Changed VERSION to 2.18.0.dev0 (#3140)
KshitijLakhani Jun 23, 2026
77054fa
[Common] Add dense router output for fused router (#3129)
harryzhou2000 Jun 23, 2026
cd20e9d
Update FE to 1.25 (#3139)
cyanguwa Jun 23, 2026
ab66858
[Common][PyTorch]Current Scaling Group Quantization + Enabling Varyin…
vthumbe1503 Jun 24, 2026
42cb81b
[PyTorch] Enable Weight Preswizzling during Quantization in TE Module…
cael-ling Jun 24, 2026
7010118
[PyTorch][CP] Add THD format support for AllGather-based Context Para…
sudhakarsingh27 Jun 25, 2026
81412e4
Avoid unpickling the extra state when not needed (#3123)
ptrendx Jun 25, 2026
2c7d5a0
Add L2 score mod distributed attention shape (#3147)
vcherepanov-nv Jun 26, 2026
43093f1
[torch.compile] Bunch of small changes needed for enabling torch.comp…
pggPL Jun 27, 2026
42f37ff
[JAX] Expert Parallelism: JAX primitives + VJPs (#3036)
phu0ngng Jun 27, 2026
ee78711
Revert "Add wheel support for Newton-Schulz method via cuSolverMp" (#…
ksivaman Jun 29, 2026
a076917
[PyTorch][torch.compile] Decouple amax reduction group from the quant…
pggPL Jun 29, 2026
4cd244e
[PyTorch] Expert Parallelism: PyTorch wrapper + autograd ops with sym…
phu0ngng Jun 29, 2026
90baf02
[Common] Update NCCL submodule to have the fix for MAX_SUPPORTED_TOKE…
phu0ngng Jun 30, 2026
46bdc85
[PyTorch] Preserve fprop operands for dequantized backward override (…
negvet Jun 30, 2026
353206d
[PyTorch] Make quantized-tensor __repr__ safe (#3146)
pggPL Jun 30, 2026
3df5e19
[Common] EP C API: version config structs and extend `nvte_ep_prepare…
phu0ngng Jun 30, 2026
25ad5cb
Graph Safe Current Scaling Support for GroupedLinear Module/Ops + Fix…
vthumbe1503 Jul 1, 2026
4cd705b
[PyT] [Common] add support for enabling cuda graph under thd format i…
HaochenYuan Jul 1, 2026
9f2074e
[Common/PyTorch] Grouped-quantize kernels for 1D and 2D FP8 block-sca…
denera Jul 1, 2026
3558abb
[JAX] Keep the routing map format alive and EP multiprocess tests in …
KshitijLakhani Jul 1, 2026
f721112
docs: document attention backend selection (#3142)
sbhavani Jul 2, 2026
dc57958
Skip MXFP8 MFSDP tests on hopper (#3163)
vthumbe1503 Jul 3, 2026
7cb8b31
[Common] Blackwell skip condition for C++ grouped FP8 block-scaling t…
denera Jul 3, 2026
75609ae
Disable cuDNN 9.23.0/9.23.1 for MXFP8 attention (#3173)
cyanguwa Jul 6, 2026
e201f22
[Pytorch][Bug] Requires Grad doesnt flow through Autograd boundaries …
vthumbe1503 Jul 6, 2026
f99a017
[PyTorch][torch.compile] Make quantizers opaque value objects (#3152)
pggPL Jul 6, 2026
8d06ed4
Remove cuDNN frontend submodule (#3169)
vcherepanov-nv Jul 7, 2026
50bd21b
[Common/PyTorch] Guard FP8 per-tensor scaling grouped GEMM on Hopper …
pggPL Jul 7, 2026
20a4912
[ALL] Support int32 topk_idx in TE EP (#3185)
phu0ngng Jul 8, 2026
8b99682
docs(readme): update latest news (#3121)
sbhavani Jul 8, 2026
01bbe3e
Fix the numerical issue with the Fused CE loss (#3193)
ptrendx Jul 9, 2026
930c8e4
Work around nvcc 13.0 parse failure in group_quantize_fp8.cuh (#3194)
xiuhu17 Jul 9, 2026
3bc89b9
Install cuDNN frontend in GitHub builds (#3196)
sudhakarsingh27 Jul 9, 2026
645c4af
[PyTorch] NVFP4: emit GEMM-swizzled scales in the non-RHT 2D quantize…
cael-ling Jul 9, 2026
7dbdf7b
Remove unused fused attention workspace optimization control (#3180)
ptrendx Jul 10, 2026
dfa0744
Enable NVFP4 RHT amax for grouped SReLU MLP (#3133)
sraman-rgb Jul 10, 2026
b782ea6
TE EP integration to MoEBlock (#3116)
tdophung Jul 10, 2026
6377ca1
[PyTorch] Fix GIL/refcount abort in Comm+GEMM overlap and NCCL-EP ini…
denera Jul 10, 2026
aef96db
Migrate norms and softmax kernels to NVRTC (#3156)
CarlosGomes98 Jul 14, 2026
215e15e
[Common] Support scaled & clamped swiglu, srelu for BF16 (#3132)
zhongbozhu Jul 14, 2026
70957ad
[Common] Pass cu_seqlens and token-unit ragged offsets directly to cu…
egilliam-nv Jul 14, 2026
bfdf24d
[PyTorch] Add per-version FlashAttention env vars (NVTE_FLASH_ATTN_V2…
wdykas Jul 15, 2026
eabdd46
Update list of authorized CI users (#3211)
timmoon10 Jul 15, 2026
9d92fa0
Fix Cuda Graph based MOE Tests Hang in CI (#3210)
vthumbe1503 Jul 15, 2026
68493d2
[PyTorch] Add optional caller-provided output/grad-input buffers to …
phu0ngng Jul 17, 2026
868d8d9
Fix FusedAdam empty tensor handling (#3212)
wujingyue Jul 20, 2026
3a49d65
Merge upstream NVIDIA TransformerEngine release v2.18 into ROCm dev
matthiasdiener Aug 27, 2026
0b1399e
[ROCm] IFU v2.18: Resolve merge conflicts and adapt for ROCm
matthiasdiener Aug 27, 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
3 changes: 1 addition & 2 deletions .clang-format
Original file line number Diff line number Diff line change
Expand Up @@ -261,7 +261,7 @@ SpacesInParensOptions:
InEmptyParentheses: false
Other: false
SpacesInSquareBrackets: false
Standard: Auto
Standard: c++17
StatementAttributeLikeMacros:
- Q_EMIT
StatementMacros:
Expand All @@ -277,4 +277,3 @@ WhitespaceSensitiveMacros:
- PP_STRINGIZE
- STRINGIZE
...

8 changes: 4 additions & 4 deletions .github/workflows/build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ jobs:
run: |
apt-get update
apt-get install -y git python3.9 pip cudnn9-cuda-12
pip install cmake==3.21.0 pybind11[global] ninja
pip install cmake==3.21.0 pybind11[global] ninja "nvidia-cudnn-frontend>=1.25.0"
- name: 'Checkout'
uses: actions/checkout@v3
with:
Expand Down Expand Up @@ -77,7 +77,7 @@ jobs:
docker exec builder bash -c '\
apt-get update && \
apt-get install -y git python3.9 pip cudnn9-cuda-12 && \
pip install cmake torch ninja pydantic importlib-metadata>=1.0 packaging pybind11 numpy einops onnxscript && \
pip install cmake torch ninja pydantic importlib-metadata>=1.0 packaging pybind11 numpy einops onnxscript "nvidia-cudnn-frontend>=1.25.0" && \
apt-get clean \
'

Expand All @@ -96,7 +96,7 @@ jobs:
options: --user root
steps:
- name: 'Dependencies'
run: pip install cmake==3.21.0 pybind11[global]
run: pip install cmake==3.21.0 pybind11[global] "nvidia-cudnn-frontend>=1.25.0"
- name: 'Checkout'
uses: actions/checkout@v3
with:
Expand Down Expand Up @@ -148,7 +148,7 @@ jobs:
- name: 'Dependencies'
run: |
docker exec builder bash -c '\
pip install cmake==3.21.0 pybind11[global] einops onnxscript && \
pip install cmake==3.21.0 pybind11[global] einops onnxscript "nvidia-cudnn-frontend>=1.25.0" && \
pip install torch --no-cache-dir --index-url https://download.pytorch.org/whl/cu130
'
- name: 'Build'
Expand Down
2 changes: 2 additions & 0 deletions .github/workflows/trigger-ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,8 @@ jobs:
|| github.actor == 'kainzhong'
|| github.actor == 'cspades'
|| github.actor == 'jomitchellnv'
|| github.actor == 'fheinecke'
|| github.actor == 'janekb04'
)
steps:
- name: Check if comment is issued by authorized person
Expand Down
3 changes: 0 additions & 3 deletions .gitmodules
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,6 @@
path = examples/pytorch/minGPT
url = https://github.com/HaiShaw/minGPT.git
branch = transformer-engine
[submodule "3rdparty/cudnn-frontend"]
path = 3rdparty/cudnn-frontend
url = https://github.com/NVIDIA/cudnn-frontend.git
[submodule "3rdparty/aotriton"]
path = 3rdparty/aotriton
url = https://github.com/ROCm/aotriton.git
Expand Down
1 change: 0 additions & 1 deletion 3rdparty/cudnn-frontend
Submodule cudnn-frontend deleted from 8d19d3
2 changes: 1 addition & 1 deletion 3rdparty/nccl
Submodule nccl updated 176 files
1 change: 1 addition & 0 deletions MANIFEST.in
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
recursive-include transformer_engine/common/include *.*
recursive-include transformer_engine/common/cmake *.cmake
recursive-include build_tools *.py *.txt
5 changes: 5 additions & 0 deletions README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -347,6 +347,11 @@ Transformer Engine
Latest News
===========

* [06/2026] `Boosting MoE Training Throughput with Advanced Fusion Kernels <https://developer.nvidia.com/blog/boosting-moe-training-throughput-with-advanced-fusion-kernels/>`_
* [06/2026] `Nemotron 3 Ultra: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoning <https://research.nvidia.com/labs/nemotron/files/NVIDIA-Nemotron-3-Ultra-Technical-Report.pdf>`_
* [06/2026] `Train Models Faster with JAX and MaxText Using NVFP4 on NVIDIA Blackwell <https://developer.nvidia.com/blog/train-models-faster-with-jax-and-maxtext-using-nvfp4-on-nvidia-blackwell/>`_
* [04/2026] `Run High-Throughput Reinforcement Learning Training with End-to-End FP8 Precision <https://developer.nvidia.com/blog/run-high-throughput-reinforcement-learning-training-with-end-to-end-fp8-precision/>`_
* [02/2026] `Using NVFP4 Low-Precision Model Training for Higher Throughput Without Losing Accuracy <https://developer.nvidia.com/blog/using-nvfp4-low-precision-model-training-for-higher-throughput-without-losing-accuracy/>`_
* [12/2025] `NVIDIA Nemotron 3: Efficient and Open Intelligence <https://arxiv.org/abs/2512.20856>`_ - trained with NVFP4 on Transformer Engine
* [11/2025] `NVIDIA Blackwell Architecture Sweeps MLPerf Training v5.1 Benchmarks <https://developer.nvidia.com/blog/nvidia-blackwell-architecture-sweeps-mlperf-training-v5-1-benchmarks/>`_
* [11/2025] `Scale Biology Transformer Models with PyTorch and NVIDIA BioNeMo Recipes <https://developer.nvidia.com/blog/scale-biology-transformer-models-with-pytorch-and-nvidia-bionemo-recipes/>`_
Expand Down
6 changes: 0 additions & 6 deletions benchmarks/attention/benchmark_attention.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,6 @@
num_iters = 3
# checkpointing
ckpt_attn = False
# workspace optimization path for cuDNN attention
workspace_opt = True
# QKV memory layout
qkv_layout = "bshd_bshd_bshd"
# padding between sequences for qkv_format=thd
Expand Down Expand Up @@ -59,7 +57,6 @@ def benchmark_dot_product_attention(model, fused_attn_supported, flash_attn_supp
"FusedAttention",
ckpt_attn,
qkv_layout,
workspace_opt,
pad_between_seqs,
is_training,
)
Expand All @@ -70,7 +67,6 @@ def benchmark_dot_product_attention(model, fused_attn_supported, flash_attn_supp
"FlashAttention",
ckpt_attn,
qkv_layout,
workspace_opt,
pad_between_seqs,
is_training,
)
Expand All @@ -90,7 +86,6 @@ def benchmark_dot_product_attention(model, fused_attn_supported, flash_attn_supp
"FusedAttention",
ckpt_attn,
qkv_layout,
workspace_opt,
pad_between_seqs,
is_training,
)
Expand All @@ -107,7 +102,6 @@ def benchmark_dot_product_attention(model, fused_attn_supported, flash_attn_supp
"FlashAttention",
ckpt_attn,
qkv_layout,
workspace_opt,
pad_between_seqs,
is_training,
)
Expand Down
4 changes: 0 additions & 4 deletions benchmarks/attention/benchmark_attention_rocm.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,8 +35,6 @@
num_iters = 10
# Checkpointing attention
ckpt_attn = False
# Workspace optimization for attention
workspace_opt = True
# QKV memory layout
qkv_layout = "bshd_bshd_bshd"
# Padding between sequences for qkv_format=thd
Expand Down Expand Up @@ -211,7 +209,6 @@ def benchmark_dot_product_attention(model, attention, column_name, dirname):
attention,
ckpt_attn,
qkv_layout,
workspace_opt,
pad_between_seqs,
is_training,
)
Expand All @@ -232,7 +229,6 @@ def benchmark_dot_product_attention_profiler(model, attention, column_name):
attention,
ckpt_attn,
qkv_layout,
workspace_opt,
pad_between_seqs,
is_training,
)
Expand Down
Loading
Loading