Commit 3102fdd
[C] Normalization Refactor + Adding CUDNN backend (NVIDIA#1315)
* cuDNN normalization integration
* TE Norm refactor
* TE Norm APIs changes.
---------
Signed-off-by: Phuong Nguyen <[email protected]>
Co-authored-by: Tim Moon <[email protected]>1 parent d8b13cb commit 3102fdd
File tree
37 files changed
+3029
-3727
lines changed- tests/cpp/operator
- transformer_engine
- common
- include/transformer_engine
- layer_norm
- normalization
- layernorm
- rmsnorm
- rmsnorm
- jax
- cpp_extensions
- csrc
- extensions
- paddle/csrc
- pytorch/csrc
- extensions
37 files changed
+3029
-3727
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
10 | 10 | | |
11 | 11 | | |
12 | 12 | | |
13 | | - | |
14 | | - | |
| 13 | + | |
15 | 14 | | |
16 | 15 | | |
17 | 16 | | |
| |||
This file was deleted.
0 commit comments