Skip to content

Commit c0643e5

Browse files
build: AUT-1613 bump transformer-engine to release_v2.18 tip (#5705)
Signed-off-by: svcnemo-autobot <svcnemo-autobot@nvidia.com>
1 parent 6124fbe commit c0643e5

2 files changed

Lines changed: 12 additions & 7 deletions

File tree

‎pyproject.toml‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ override-dependencies = [
130130
"torch; sys_platform == 'never'",
131131
"torchvision; sys_platform == 'never'",
132132
"triton; sys_platform == 'never'",
133-
"transformer-engine @ git+https://github.com/NVIDIA/TransformerEngine.git@4329ff84bfbdaa778a33cba02a15fb0807c64689",
133+
"transformer-engine @ git+https://github.com/NVIDIA/TransformerEngine.git@27486e03cfc1fa41f6932dcecdc47c71c47eac3e",
134134
"mlflow>=3.15.1", # To address CVE-2025-15031
135135
"cachetools>=5.0.0",
136136
"cryptography>=43.0.0,<47",
@@ -169,11 +169,12 @@ requires-dist = []
169169
# above) changes — an out-of-date entry sends uv back to building TE.
170170
[[tool.uv.dependency-metadata]]
171171
name = "transformer-engine"
172-
version = "2.17.1+4329ff84"
172+
version = "2.18.0+27486e03"
173173
requires-dist = [
174174
"einops",
175175
"importlib-metadata>=1.0",
176176
"nvdlfw-inspect",
177+
"nvidia-cudnn-frontend>=1.25.0",
177178
"onnx",
178179
"onnxscript",
179180
"packaging",
@@ -242,6 +243,7 @@ build = [
242243
"numpy",
243244
"ninja",
244245
"nvidia-mathdx",
246+
"nvidia-cudnn-frontend>=1.25.0",
245247
]
246248
# CVE-carrying WAN diffusion codecs, needed only by WAN video caching
247249
# (src/megatron/bridge/diffusion/models/wan/inference/utils.py). All three are suppressed

‎uv.lock‎

Lines changed: 8 additions & 5 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)