Skip to content

Commit 3102fdd

Browse files
phu0ngngtimmoon10
andauthored
[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

37 files changed

+3029
-3727
lines changed

tests/cpp/operator/CMakeLists.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,7 @@ add_executable(test_operator
1010
test_cast_transpose_dbias_dgelu.cu
1111
test_cast_transpose_dgeglu.cu
1212
test_act.cu
13-
test_layernorm.cu
14-
test_rmsnorm.cu
13+
test_normalization.cu
1514
test_multi_cast_transpose.cu
1615
test_multi_padding.cu
1716
test_causal_softmax.cu

tests/cpp/operator/test_layernorm.cu

Lines changed: 0 additions & 302 deletions
This file was deleted.

0 commit comments

Comments
 (0)