ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-16 08:44:55 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
3696fe1c76f50b2d16742632128c41f7869bc119
composable_kernel/example
History
rocking 3696fe1c76 Layernorm and groupnorm support to save mean and inverse std in forward (#929) 
* save mean and inverse std in normalization

* Save mean and inverse std in splitK

* Vector save mean and inv std

* Modify instance for save mean and std

* simplify the layernorm example

* Save mean and std in groupnorm example

* Save mean and inv std in ckProfiler and test

* Remove compute data type from base class

* Save mean and inv std in client example

* Add changelog

* clang format

* Fix compile error

* Refine naming

* Avoid error in bf16

* revert changelog
2023-10-19 07:36:29 +08:00
..
01_gemm
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
02_gemm_bilinear
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
03_gemm_bias_relu
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
04_gemm_add_add_fastgelu
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
09_convnd_fwd
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
10_convnd_fwd_multiple_d_multiple_reduce
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
12_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
13_pool2d_fwd
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
14_gemm_quantization
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
15_grouped_gemm
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
16_gemm_multi_d_multi_reduces
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
17_convnd_bwd_data
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
18_batched_gemm_reduce
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
19_binary_elementwise
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
20_grouped_conv_bwd_weight
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
21_gemm_layernorm
Layernorm and groupnorm support to save mean and inverse std in forward (#929)
2023-10-19 07:36:29 +08:00
22_cgemm
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
23_softmax
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
24_batched_gemm
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
25_gemm_bias_e_permute
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
26_contraction
Revert "Add support for mixed precision in contraction scale and bilinear" (#967)
2023-10-05 14:58:23 -07:00
27_layernorm
Layernorm and groupnorm support to save mean and inverse std in forward (#929)
2023-10-19 07:36:29 +08:00
28_grouped_gemm_bias_e_permute
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
29_batched_gemm_bias_e_permute
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
30_grouped_conv_fwd_multiple_d
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
31_batched_gemm_gemm
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
32_batched_gemm_scale_softmax_gemm
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
33_multiple_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
34_batchnorm
Batchnorm splitk single kernel (#771)
2023-07-06 10:58:55 -05:00
35_splitK_gemm
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
36_sparse_embedding
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
37_batched_gemm_add_add_relu_gemm_add
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
38_grouped_conv_bwd_data_multiple_d
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
39_permute
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
40_conv2d_fwd_quantization
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
41_grouped_conv_conv_fwd
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
42_groupnorm
Layernorm and groupnorm support to save mean and inverse std in forward (#929)
2023-10-19 07:36:29 +08:00
43_splitk_gemm_bias_e_permute
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
44_elementwise_permute
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
45_elementwise_normalization
Layernorm and groupnorm support to save mean and inverse std in forward (#929)
2023-10-19 07:36:29 +08:00
46_gemm_add_multiply
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
47_gemm_bias_softmax_gemm_permute
Enable gfx941 and gfx942 architectures. (#752)
2023-06-15 08:20:59 -07:00
48_pool3d_fwd
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
49_maxpool2d_bwd
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
50_put_element
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
51_avgpool3d_bwd
Average pool backward deviceOP and example (#797)
2023-08-10 12:04:35 +08:00
52_im2col_col2im
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
60_gemm_multi_ABD
Add contraction_multi_abd (#972)
2023-10-17 20:17:58 -05:00
61_contraction_multi_ABD
Add contraction_multi_abd (#972)
2023-10-17 20:17:58 -05:00
CMakeLists.txt
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00