- 07 Oct, 2024 1 commit
-
-
Xiaowei Ren authored
* change API for hierarchical CP Signed-off-by:
Xiaowei Ren <xren@nvidia.com> * move fp8 code before qkv reshape Signed-off-by:
pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
-
- 04 Oct, 2024 1 commit
-
-
Tim Moon authored
* CPU perf optimization in linear autograd function Avoid enable_grad context when possible in cast function. Cache distributed group properties. Signed-off-by:
Tim Moon <tmoon@nvidia.com> * CPU perf optimization in prepare_forward function Avoid torch.nn.Module impl of __setattr__. Signed-off-by:
-
- 03 Oct, 2024 1 commit
-
-
Charlene Yang authored
move block_table arg to varlen_func section Signed-off-by:Charlene Yang <8636796+cyanguwa@users.noreply.github.com>
-
- 01 Oct, 2024 2 commits
-
-
Przemyslaw Tredak authored
* Removing the unused options from GroupedLinear docs and fixing the bug with offsets Signed-off-by:
Przemyslaw Tredak <ptredak@nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * offsets -> fp8_meta_offsets Signed-off-by:
-
Kirthi Shankar Sivamani authored
Add pool argument to make_graphed_callable Signed-off-by:Kirthi Shankar Sivamani <ksivamani@nvidia.com>
-
- 27 Sep, 2024 2 commits
-
-
Charlene Yang authored
* fix detection of 3 in 3hd/h3d layouts Signed-off-by:
-
Paweł Gadziński authored
* Docs fixes Signed-off-by:
Pawel Gadzinski <pgadzinski@nvidia.com> * docs fix Signed-off-by:
-
- 20 Sep, 2024 1 commit
-
-
Przemyslaw Tredak authored
* Restore compatibility with Python 3.8 Signed-off-by:
-
- 19 Sep, 2024 1 commit
-
-
Xin Yao authored
* relax contiguous check for flash attention Signed-off-by:
Xin Yao <xiny@nvidia.com> * force contiguous for cp Signed-off-by:
-
- 18 Sep, 2024 2 commits
-
-
Sudhakar Singh authored
* make rotary_base arg Signed-off-by:
Sudhakar Singh <sudhakars@nvidia.com> * rotary base can be a float Co-authored-by:
Tim Moon <4406448+timmoon10@users.noreply.github.com> Signed-off-by:
-
Alp Dener authored
* Check if network interface name is valid and show useful warning message when initializing Userbuffers Signed-off-by:
Alp Dener <adener@nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Fix formatting issue in warning message. Co-authored-by:
-
- 16 Sep, 2024 1 commit
-
-
Jason Wang authored
add dtensor support for te optimizers Signed-off-by:
jasonwan <jasonwan@nvidia.com> Co-authored-by:
-
- 11 Sep, 2024 1 commit
-
-
Tim Moon authored
* Add base class for tensor proxies Signed-off-by:
-
- 09 Sep, 2024 2 commits
-
-
Xiaowei Ren authored
* clean code for CP function args Signed-off-by:
-
Xin Yao authored
* propagate scale_inv modification to GroupedLinear Signed-off-by:
-
- 05 Sep, 2024 3 commits
-
-
Ruibin Cheung authored
* [TE/PyTorch][MoE] Add FP8 padding and unpadding module 1. Add multi-tensor padding kernel for FP8 with padding size = 16. 2. Add FP8Padding and Fp8Unpadding module 3. Add Padded GroupedLinear unit tests --------- Signed-off-by:
beinggod <zhangruibin@01.ai> Co-authored-by:
Phuong Nguyen <36155692+phu0ngng@users.noreply.github.com>
-
Selvaraj Anandaraj authored
* Added offloading support FP8 attention Signed-off-by:
Selvaraj Anandaraj <selvaraja@login-eos02.eos.clusters.nvidia.com> * Update transformer_engine/pytorch/attention.py Co-authored-by:
Selvaraj Anandaraj <anandaraj@wisc.edu> * Fix Signed-off-by:
-
Xin Yao authored
* fp8 mha with rope Signed-off-by:
-
- 03 Sep, 2024 2 commits
-
-
Ruibin Cheung authored
[PyTorch] Add contiguous check for grouped gemm Signed-off-by:
-
Kirthi Shankar Sivamani authored
* Improvements for wheels Signed-off-by:
-
- 30 Aug, 2024 2 commits
-
-
Xiaowei Ren authored
* fix qkv_dtype of FP8+CP Signed-off-by:
-
Charlene Yang authored
* fix FP8 logic when FA3 is not installed Signed-off-by:
-
- 29 Aug, 2024 1 commit
-
-
Xin Yao authored
* remove dtype from args * update docs with permutation ops --------- Signed-off-by:
-
- 23 Aug, 2024 2 commits
-
-
Charlene Yang authored
* WIP: add fa3 Signed-off-by:
-
Alexandros Koumparoulis authored
Signed-off-by:Alexandros Koumparoulis <akoumparouli@nvidia.com>
-
- 22 Aug, 2024 1 commit
-
-
NVJiangShao authored
* Add permutation functions * Add permutation ops * Remove the dependency on cutlass * Move permutation.py out of module dir * Rewrite the unit test and enable skipping if FP8 is unavailable * Rename exposed C++ API and reorder its parameters + take NVTETensor as inputs * Use Float8Tensor for FP8 input * Move dtype to ctx --------- Signed-off-by:
Jiang Shao <jiangs@nvidia.com> Co-authored-by:
Qi Zhang <qizhang@nvidia.com> Co-authored-by:
-
- 21 Aug, 2024 3 commits
-
-
Charlene Yang authored
* add support for padding in UnfusedDPA Signed-off-by:
-
Xiaowei Ren authored
* add window_size to AttnFuncWithCP Signed-off-by:
Xiaowei Ren <xren@cs-cw-dfw-login-01.cm.cluster>
-
Tim Moon authored
* Perform scale-inv update in cast-transpose kernels Signed-off-by:
-
- 20 Aug, 2024 1 commit
-
-
hXl3s authored
feat(pytorch): Allow TransformerLayer and MultiheadAttention to accept sequence length parameters (#1066) * Added ability for seqlen for transformer and mha layer Signed-off-by:
Lukasz Pierscieniewski <lukaszp@nvidia.com> * Documentation for new parameters Signed-off-by:
-
- 16 Aug, 2024 2 commits
-
-
Shijie authored
* support dtype casting fusion in FusedAdam Signed-off-by:
Shijie Wang <jaywan@nvidia.com> * minor changes Signed-off-by:
-
Xiaowei Ren authored
* add window_size to AttnFuncWithCP Signed-off-by:
-
- 15 Aug, 2024 2 commits
-
-
Charlene Yang authored
fix typos regarding t in thd Signed-off-by: -
Marks101 authored
Signed-off-by:
Markus Schnoes <markus.schnoes@gmx.de> Co-authored-by:
-
- 13 Aug, 2024 2 commits
-
-
Charlene Yang authored
* merge k_channels and v_channels back to kv_channels and accept a tuple Signed-off-by:
-
vasunvidia authored
* DGRAD-RS overlap bug fix This PR fixes a bug in enabling DGRAD-RS overlap by adding the layer to the correct method list. Previously, the RS-DGRAD overlap layer was incorrectly added to pipeline method list even if ring_exchange method is specified in config. Signed-off-by:
Vasudevan Rengasamy <vrengasamy@nvidia.com> * Bug fix for ring_exchange ReduceScatter ring_exchange RS uses main_stream for last GEMM chunk. But the send/recv streams wait for stream_compute during last chunk. Signed-off-by:
-
- 12 Aug, 2024 1 commit
-
-
vasunvidia authored
Buf fix for num_warmup_iters=0 case Signed-off-by:
-
- 10 Aug, 2024 1 commit
-
-
Tim Moon authored
* Add op for in-place add Signed-off-by:
-
- 09 Aug, 2024 2 commits
-
-
Xin Yao authored
* use fused_multi_cast_transpose Signed-off-by:
-
Alp Dener authored
[C/PyTorch] Fixed incorrect use of `torch.distributed.new_group()` when creating intra-node group in `initialize_ub()` (#1087) * updated initialize_ub() to use new_subgroups_by_enumeration() to generate intra-node groups, added new unit tests for TE layers with comm overlap Signed-off-by:
-
