Commit 7df9094
File tree
639 files changed
+10303
-2885
lines changed- .github
- actions/build-pytorch-wheel
- scripts
- workflows
- benchmarks
- attention
- linear
- build_tools
- wheel_utils
- docs
- _static/css
- api
- c
- debug
- examples
- attention
- onnx
- te_gemma
- te_llama
- getting_started
- examples
- jax
- collective_gemm
- encoder
- mnist
- pytorch
- comm_gemm_overlap
- fsdp
- mnist
- qa
- L0_cppunittest
- L0_jax_distributed_unittest
- L0_jax_lint
- L0_jax_unittest
- L0_jax_wheel
- L0_license
- L0_pytorch_debug_unittest
- L0_pytorch_lint
- L0_pytorch_unittest
- L0_pytorch_wheel
- L1_cpp_distributed
- L1_jax_distributed_unittest
- L1_pytorch_distributed_unittest
- L1_pytorch_mcore_integration
- L1_pytorch_onnx_unittest
- L1_pytorch_thunder_integration
- L2_jax_distributed_unittest
- L2_jax_unittest
- L3_pytorch_FA_versions_test
- tests
- cpp_distributed
- cpp
- operator
- util
- jax
- pytorch
- attention
- debug
- distributed
- layernorm_mlp
- nvfp4
- references
- transformer_engine
- common
- activation
- cast
- core
- dispatch
- fp8
- mxfp8
- specialized
- nvfp4
- comm_gemm_overlap
- userbuffers
- comm_gemm
- dropout
- fused_attn
- fused_rope
- fused_router
- fused_softmax
- gemm
- hadamard_transform
- include/transformer_engine
- multi_tensor
- normalization
- layernorm
- rmsnorm
- nvshmem_api
- permutation
- recipe
- swizzle
- transpose
- rtc
- triton
- util
- debug
- features
- utils
- pytorch
- jax
- cpp_extensions
- csrc
- extensions
- flax
- quantize
- triton_extensions
- pytorch
- attention
- dot_product_attention
- cpp_extensions
- csrc
- extensions
- multi_tensor
- custom_recipes
- module
- ops
- basic
- fused
- optimizers
- tensor
- storage
- triton
Some content is hidden
Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.
639 files changed
+10303
-2885
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
| 1 | + | |
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
| 1 | + | |
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | 1 | | |
2 | 2 | | |
3 | | - | |
| 3 | + | |
4 | 4 | | |
5 | 5 | | |
6 | 6 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | 1 | | |
2 | 2 | | |
3 | | - | |
| 3 | + | |
4 | 4 | | |
5 | 5 | | |
6 | 6 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
| 1 | + | |
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
| 1 | + | |
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
| 1 | + | |
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
| 1 | + | |
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
| 1 | + | |
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
| |||
17 | 17 | | |
18 | 18 | | |
19 | 19 | | |
20 | | - | |
| 20 | + | |
21 | 21 | | |
22 | 22 | | |
23 | 23 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
| 1 | + | |
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
| |||
0 commit comments