- 09 Mar, 2026 2 commits
-
-
wenjh authored
Signed-off-by:wenjh <wenjh@sugon.com>
-
wuyf1 authored
Signed-off-by:
Wuyufan <Wuyf1@sugon.com> 解决了如下问题： 1. FA 48个单测报错问题（fA输入:[B,S,H,D] reshape to [blocknums,blocksize,H,D]）. 2. 在解决FA 报错的时候一个kernel launch error. 3. norm_mlp问题暂时用rest_rng_state解决 See merge request dcutoolkit/deeplearing/TransformerEngine!77 Co-authored-by:
Tangao <2205747538@qq.com> Co-authored-by:
wuyufffan <1095978552@qq.com>
-
- 03 Mar, 2026 2 commits
- 24 Feb, 2026 1 commit
-
-
wenjh authored
Signed-off-by:
-
- 18 Feb, 2026 1 commit
-
-
Przemek Tredak authored
Signed-off-by:Przemek Tredak <ptredak@nvidia.com>
-
- 17 Feb, 2026 2 commits
-
-
jberchtold-nvidia authored
* initial debug of inspect ffi Signed-off-by:
Jeremy Berchtold <jberchtold@nvidia.com> * writing binary dumps of tensors works Signed-off-by:
pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
-
Hemil Desai authored
- Include `build_tools/` in the source distribution via `MANIFEST.in` so that cached builds from `uv` (and `pip`) can resolve `setup.py`'s top-level imports `setup.py` imports from `build_tools` at the top level: ```python from build_tools.build_ext import CMakeExtension, get_build_ext from build_tools.te_version import te_version from build_tools.utils import cuda_archs, cuda_version, ... ``` The `__legacy__` build backend in `pyproject.toml` adds the source root to `sys.path`, so these imports work when building directly from the source tree. However, `build_tools/` is not included in the sdist because: 1. `MANIFEST.in` did not list it 2. `build_tools/` is not discovered by `find_packages()` (it's a standalone directory at the repo root, not under `transformer_engine/`) When `uv` caches the sdist and later builds a wheel from it, the sdist is extracted to a temporary directory where `build_tools/` is absent, causing a...
-
- 13 Feb, 2026 2 commits
-
-
Teddy Do authored
* [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * adding more stuff missing from cherry picky jeremy PR for inspecting Signed-off-by:
tdophung <tdophung@nvidia.com> * fix some tracing issues when intergating to maxtext Signed-off-by:
JAX Toolbox <jax@nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * remove Jeremy's PR for inspect ffi Signed-off-by:
-
Kirthi Shankar Sivamani authored
* Make grouped weights opt-in Signed-off-by:
Kirthi Shankar Sivamani <ksivamani@nvidia.com> * Change varname Signed-off-by:
-
- 12 Feb, 2026 6 commits
-
-
vcherepanov-nv authored
* Remove nvshmem usage Signed-off-by:
Vladimir Cherepanov <vcherepanov@nvidia.com> * Renamings Signed-off-by:
-
Harikrishna KP authored
fix: correct copy-paste error messages in FusedSGD Signed-off-by:Mr-Neutr0n <64578610+Mr-Neutr0n@users.noreply.github.com>
-
Kim, Jin (Jay@SKT) authored
* Add sigmoid GLU Signed-off-by:
Kim, Jin <jinn.kim@sk.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Signed-off-by:
Tim Moon <tmoon@nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Fix incorrect reshape Signed-off-by:
Tim Moon <4406448+timmoon10@users.noreply.github.com> * Apply suggestion from @timmoon10 Signed-off-by:
-
Tim Moon authored
* Add ops for MoE grouped MLP Signed-off-by:
-
Oleg Goncharov authored
* Added GEMM-ready preswizzling option Signed-off-by:
Oleg Goncharov <ogoncharov@nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Signed-off-by:
-
Santosh Bhavani authored
fix: handle nvidia namespace packages where __file__ is None Signed-off-by:
Santosh Bhavani <santosh.bhavani@live.com> Co-authored-by:
-
- 11 Feb, 2026 4 commits
-
-
Kirthi Shankar Sivamani authored
* NVFP4 GroupedQuantize Signed-off-by:
Zhongbo Zhu <zhongboz@nvidia.com> Co-authored-by:
-
Kirthi Shankar Sivamani authored
* PyTorch-Python GroupedTensor Signed-off-by:
greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com> Signed-off-by:
-
Lifu Zhang authored
* Fix on TE to support Mcore Vision Encoder CUDA Graph Signed-off-by:
Lifu Zhang <lifuz@login-lyris02.lyris.clusters.nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * refactoring code Signed-off-by:
-
Faradawn Yang authored
* fix broken link of quickstart guide Signed-off-by:
Faradawn Yang <73060648+faradawn@users.noreply.github.com> * Update README.rst Co-authored-by:
Paweł Gadziński <62263673+pggPL@users.noreply.github.com> Signed-off-by:
-
- 10 Feb, 2026 2 commits
-
-
Jacket authored
Signed-off-by:Kaining Zhong <kainingz@nvidia.com>
-
Przemyslaw Tredak authored
* Fix the compilation warnings for the PyTorch extension Signed-off-by:
Przemyslaw Tredak <ptrendx@gmail.com> --------- Signed-off-by:
-
- 09 Feb, 2026 3 commits
-
-
Pingtian Li authored
* add grad reduce api for cuda graph hook Signed-off-by:
Pingtian Li <pingtianl@nvidia.com> * fix code consistency Signed-off-by:
-
Paweł Gadziński authored
fix Signed-off-by:Pawel Gadzinski <pgadzinski@nvidia.com>
-
jberchtold-nvidia authored
* expand troubleshooting docs Signed-off-by:
jberchtold-nvidia <158520091+jberchtold-nvidia@users.noreply.github.com> * Update README.rst Co-authored-by:
-
- 07 Feb, 2026 1 commit
-
-
Charlene Yang authored
bucket max_b with more granularity when >512 Signed-off-by:Charlene Yang <8636796+cyanguwa@users.noreply.github.com>
-
- 06 Feb, 2026 2 commits
-
-
Oleg Goncharov authored
* Rebased to main Signed-off-by:
vthumbe1503 <vthumbe@nvidia.com>
-
Jacket authored
Signed-off-by:
-
- 04 Feb, 2026 3 commits
-
-
wenjh authored
Signed-off-by: -
wenjh authored
Signed-off-by: -
jberchtold-nvidia authored
* Update README.rst Signed-off-by:
-
- 03 Feb, 2026 3 commits
-
-
Paweł Gadziński authored
* init Signed-off-by:
-
Oleg Goncharov authored
* Fixed scaling-factor computation for FP32 to match the reference implementation. Signed-off-by:
-
Vadim Markovtsev authored
* Support building with headers from nvidia wheels There are two changes: 1. `import nvidia` returns a namespace package with `__file__` equal to `None` 2. Add the way to force headers from nvidia wheels. Without that envvar, it's practically impossible with CUDA installed system-wide. I successfully built the package with torch using the following `uv` configuration: ``` [tool.uv.extra-build-dependencies] "transformer-engine-torch" = [ "ninja", "nvidia-cuda-crt==13.0.88", "nvidia-cuda-cccl==13.0.85", { requirement = "torch", match-runtime = true }, { requirement = "pytorch-triton", match-runtime = true }, { requirement = "nvidia-cusolver", match-runtime = true }, { requirement = "nvidia-curand", match-runtime = true }, { requirement = "nvidia-cublas", match-runtime = true }, { requirement = "nvidia-cusparse", match-runtime = true }, { requirement = "nvidia-cudnn-cu13", match-runtime = true }, { requirement = "nvidia-nvtx", match-runtime = true }, { requirement = "nvidia-cuda-nvrtc", match-runtime = true }, { requirement = "nvidia-cuda-runtime", match-runtime = true }, ] ``` Signed-off-by:Vadim Markovtsev <vadim@poolside.ai> * Apply suggestion from @ksivaman Signed-off-by:
-
- 02 Feb, 2026 1 commit
-
-
Paweł Gadziński authored
* Code drop: Update recipes documentation and remove custom recipes from low precision training Signed-off-by:
-
- 30 Jan, 2026 2 commits
-
-
Paweł Gadziński authored
* version change Signed-off-by:
-
wenjh authored
Signed-off-by:
-
- 29 Jan, 2026 1 commit
-
-
Paweł Gadziński authored
fix wheel Signed-off-by:
-
- 28 Jan, 2026 2 commits
-
-
Paweł Gadziński authored
* code drop Signed-off-by:
-
Paweł Gadziński authored
* jjit bug fix Signed-off-by:
-
