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
7ed9e9a34870f0cb4f4047b81480ca406b997088
composable_kernel/example
History
Shaojie WANG 7ed9e9a348 example of conv bwd weight 1d/2d/3d fp32/fp16/bf16 xdl (#244) 
* enable example of conv 1d/3d for bwd weight

* make bf16 kernel do not use atomic add

* using new gridwise gemm for bwd weight on convnd bwd weight

Co-authored-by: Chao Liu <chao.liu2@amd.com>

[ROCm/composable_kernel commit: ac543313bf]
2022-05-20 17:20:10 -05:00
..
01_gemm
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
02_gemm_alpha_beta
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
03_gemm_bias_relu
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
04_gemm_bias_relu_add
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
06_conv2d_fwd_bias_relu
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
07_conv2d_fwd_bias_relu_add
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
09_convnd_fwd
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
10_conv2d_bwd_data
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
11_conv2d_bwd_weight
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
12_reduce
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
13_pool2d_fwd
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
14_gemm_xdl_requant_relu_requant
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
15_grouped_gemm
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
16_gemm_reduce
Gemm reduce max (#209)
2022-05-19 21:56:56 -05:00
17_convnd_bwd_data_xdl
Validate examples in CI (#233)
2022-05-13 16:54:44 -05:00
18_batched_gemm_reduce
Gemm reduce max (#209)
2022-05-19 21:56:56 -05:00
19_binary_elementwise
Hotfix eltiwseop (#242)
2022-05-19 22:02:06 -05:00
20_convnd_bwd_weight_xdl
example of conv bwd weight 1d/2d/3d fp32/fp16/bf16 xdl (#244)
2022-05-20 17:20:10 -05:00
CMakeLists.txt
example of conv bwd weight 1d/2d/3d fp32/fp16/bf16 xdl (#244)
2022-05-20 17:20:10 -05:00