ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
c18224799926da71ef4ae8bc65761dc9f5918ea4
composable_kernel/example
History
rocking5566 fe367bc917 Refine layernorm naming and test code (#497) 
* Sync the naming

* Sync the test of layernorm with groupnorm

* Sync the naming

* Minor change for comment and log

* [What] Add saveMean and SaveInvVariance in the interface.
[Why] These can optimize the backward

[ROCm/composable_kernel commit: d4d1147f0a]
2022-11-02 16:57:28 -06:00
..
01_gemm
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
02_gemm_bilinear
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
03_gemm_bias_relu
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
04_gemm_add_add_fastgelu
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
09_convnd_fwd
Add Conv Forward on Navi21 for ResNet50 (#490)
2022-10-31 10:24:25 -06:00
10_convnd_fwd_multiple_d_multiple_reduce
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
12_reduce
Update to the Reduction API and instances (#476)
2022-10-25 09:39:11 -06:00
13_pool2d_fwd
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
14_gemm_xdl_quantization
Conv perlayer int8 quantization (#471)
2022-11-02 13:56:26 -06:00
15_grouped_gemm
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
16_gemm_multi_d_multi_reduces
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
17_convnd_bwd_data
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
18_batched_gemm_reduce
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
19_binary_elementwise
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
20_convnd_bwd_weight
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
21_gemm_layernorm
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
22_cgemm
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
23_softmax
Softmax client example (#396)
2022-09-06 12:22:48 -05:00
24_batched_gemm
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
25_gemm_bias_e_permute
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
26_contraction
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
27_layernorm
Refine layernorm naming and test code (#497)
2022-11-02 16:57:28 -06:00
28_grouped_gemm_bias_e_permute
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
29_batched_gemm_bias_e_permute
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
30_grouped_convnd_fwd_bias_relu_add
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
31_batched_gemm_gemm
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
32_batched_gemm_scale_softmax_gemm
Input/output permutation for fused attention (#460)
2022-10-27 14:58:20 -06:00
33_multiple_reduce
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
34_batchnorm
Batchnorm-forward implemented using welford method to calculate variance (#403)
2022-10-27 18:52:54 -06:00
35_splitK_gemm
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
36_sparse_embedding
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
37_batched_gemm_add_add_relu_gemm_add
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
38_grouped_conv_bwd_data_multiple_d
Add client example of grouped conv2d backward data (data type: fp16) (#481)
2022-11-02 16:54:41 -06:00
39_permute
Add 'Permute' device op & example (#408)
2022-09-19 21:30:25 -05:00
41_grouped_conv_conv_fwd
Refactor device op implementations into impl subdirectory. (#420)
2022-10-13 09:05:08 -05:00
42_groupnorm
Refine layernorm naming and test code (#497)
2022-11-02 16:57:28 -06:00
43_splitk_gemm_bias_e_permute
Example contraction splitk (#430)
2022-10-11 17:54:34 -05:00
44_conv2d_fwd_quant
Conv perlayer int8 quantization (#471)
2022-11-02 13:56:26 -06:00
44_elementwise_permute
Tensor permutation (#479)
2022-10-18 23:24:19 -05:00
CMakeLists.txt
Conv bwd data multiple d (#404)
2022-09-19 11:25:28 -05:00